"""
    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")