diff options
Diffstat (limited to 'synonymiser/db.py')
-rw-r--r-- | synonymiser/db.py | 34 |
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") + |