WIP: config file support

hamstertools/db.py

@@ -1,6 +1,7 @@
 from datetime import datetime
 
 from peewee import (
+    CompositeKey,
     SqliteDatabase,
     Model,
     CharField,
@@ -31,6 +32,14 @@ class HamsterActivity(Model):
         table_name = "activities"
 
 
+class HamsterTag(Model):
+    name = CharField()
+
+    class Meta:
+        database = db
+        table_name = "tags"
+
+
 class HamsterFact(Model):
     activity = ForeignKeyField(HamsterActivity, backref="facts")
     start_time = DateTimeField()
@@ -42,6 +51,16 @@ class HamsterFact(Model):
         table_name = "facts"
 
 
+class HamsterFactTag(Model):
+    fact = ForeignKeyField(HamsterFact, backref="tags")
+    tag = ForeignKeyField(HamsterTag, backref="facts")
+
+    class Meta:
+        database = db
+        table_name = "fact_tags"
+        primary_key = CompositeKey('fact', 'tag')
+
+
 class KimaiCustomer(Model):
     visible = BooleanField(default=True)
     name = CharField()