initialHEADmaster

1 files changed, 34 insertions, 0 deletions

diff --git a/synonymiser/db.py b/synonymiser/db.py
new file mode 100644
index 0000000..e8c830a
--- /dev/null
+++ b/synonymiser/db.py
@@ -0,0 +1,34 @@
+"""
+    db.py
+    copyright 2021 Richard Knight
+"""
+from peewee import *
+from playhouse.postgres_ext import *
+import config
+
+_db = PostgresqlExtDatabase(
+    config.db_name,
+    **{
+        "host": config.db_host,
+        "user": config.db_user,
+        "password": config.db_password
+    }
+)
+
+
+class BaseModel(Model):
+    class Meta:
+        database = _db
+
+
+class Word(BaseModel):
+    word = TextField()
+    offensive = BooleanField()
+    class Meta:
+        db_table = "word"
+
+
+class Synonym(BaseModel):
+    base = ForeignKeyField(Word, related_name="base_word")
+    synonym = ForeignKeyField(Word, related_name="syn_word")
+