Add commands support

This commit is contained in:
Jonatan Pålsson 2017-06-12 21:47:14 +02:00
parent bf661d9577
commit 529bcbb003
3 changed files with 59 additions and 1 deletions

13
Db.py
View file

@ -23,11 +23,16 @@ class LogMessage(BaseModel):
message_type = CharField()
message = CharField()
class Quote(BaseModel):
datetime = DateTimeField()
author = CharField()
message = CharField()
def create_tables():
database.connect()
try:
database.create_tables([LogMessage, Day, Channel])
database.create_tables([Quote, LogMessage, Day, Channel])
except OperationalError:
pass # Database already exists
@ -55,6 +60,12 @@ def add_log_message(channel, nickname, message_type, message = None):
message_type = message_type,
message = message)
def add_quote(author, message):
Quote.create(
author = author,
message = message,
datetime = datetime.datetime.now().strftime("%H:%m:%S"))
def show_all_messages():
for message in LogMessage.select():
print("<%s> %s" % (message.nickname, message.message))