diff options
author | Richard <q@1bpm.net> | 2021-03-14 15:55:50 +0000 |
---|---|---|
committer | Richard <q@1bpm.net> | 2021-03-14 15:55:50 +0000 |
commit | ccc8a42c42985b3227c2d9f3a27f967dcaa7d70e (patch) | |
tree | 4864e5a5fabed8d5883c13ac469fcf1d00b6bb04 /synonymiser/db.py | |
download | synonymiser-master.tar.gz synonymiser-master.tar.bz2 synonymiser-master.zip |
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") + |