Started on scheduling

Author: Innectic

messages.py

@@ -1,5 +1,5 @@
 from user import User
-from models import Command, session, CommandCommand, QuoteCommand, CubeCommand, SocialCommand
+from models import Command, session, CommandCommand, QuoteCommand, CubeCommand, SocialCommand, ScheduleCommand
 from asyncio import async, coroutine
 from functools import partial
 
@@ -46,6 +46,7 @@ def message_handler(self, data):
                 "quote": QuoteCommand,
                 "cube": CubeCommand,
                 "social": SocialCommand,
+                "schedule": ScheduleCommand,
             }
             if args[0][1:] in commands:
                 yield from self.send_message(

models.py

@@ -6,6 +6,7 @@
 from datetime import datetime
 from re import sub, findall
 from random import randrange
+from schedule import Schedule
 
 from user import User
 
@@ -176,6 +177,22 @@ def __call__(self, args, data=None):
         return "No social services were found on the streamer's profile."
 
 
+class ScheduleCommand(Command):
+    def __call__(self, args, data=None):
+        action = args[1]
+        interval = args[2]
+        text = args[3]
+
+        if action is "add":
+            time = interval[:-1]
+            modifer = interval[-1:]
+
+
+        elif action is "remove":
+            pass
+        else:
+            pass
+
 # #### TO BE REDONE IN USERS MODEL #### #
 
 
@@ -199,6 +216,7 @@ def add_points(self, username, amount):
                 amount=amount
             )
             session.add(c)
+            session.commit()
         else:
             # Todo add the user.
             pass
@@ -239,3 +257,12 @@ def reset_points(self, username):
             # TODO: Throw an error and tell the user that sent this bad things
             pass
         session.commit()
+
+
+class Schedule(Base):
+    __tablename__ = "scheduled"
+
+    id = Column(Integer, unique=True, primary_key=True)
+    text = Column(String)
+    interval = Column(Integer)
+    last_ran = Column(DateTime)

schedule.py

@@ -0,0 +1,52 @@
+
+from models import Scheduled
+from sqlalchemy.orm import Session
+
+from time import datetime
+
+from asyncio import coroutine, get_event_loop
+
+
+class Schedule:
+
+    loop = get_event_loop()
+
+    def add(self, text, interval):
+        session = Session()
+
+        query = session.query(Scheduled.Base).first()
+
+        if query:
+            c = Scheduled(
+                text=text,
+                interval=interval,
+                last_ran=datetime.utcnow()
+            )
+
+            session.add(c)
+            session.flush()
+            session.commit()
+        else:
+            raise Exception("WHAT ARE YOU DOING?!")
+
+    def remove(self, id):
+        session = Session()
+        query = session.query(Scheduled.Base).filter_by(id=id).first()
+
+        if query:
+            query.delete()
+            session.commit()
+        else:
+            raise Exception("That shouldn't have happened.")
+
+    @coroutine
+    def schedule(self, id, interval):
+        session = Session()
+        cur_time = 0
+
+        query = session.query(Scheduled.Base).filter_by(id=id).first()
+
+        if query:
+            pass
+        else:
+            raise Exception("Someting bad happened.")