aboutsummaryrefslogtreecommitdiff
path: root/synonymiser/db.py
blob: e8c830ab4f9d8f4014eb4e2661a6ea0815939afc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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")