from peewee import *
from playhouse.fields import CompressedField

import datetime
 
db = SqliteDatabase('test.db', pragmas=[('foreign_keys', 'ON')])

__tables__ = []
__all__ = ['create_tables', 'drop_tables']

def table(cls):
    __tables__.append(cls)
    __all__.append(cls.__name__)
    return cls

def create_tables():
    db.create_tables(__tables__, True)

def drop_tables():
    db.drop_tables(__tables__, True)

class BaseModel(Model):
    class Meta:
        database = db

@table
class Document(BaseModel):
    content = CompressedField()
    created = DateTimeField(default=datetime.datetime.now)
    description = CharField(null=True)
    direction = BooleanField(null=True)
    original_path = CharField(null=True)

@table
class Prefix(BaseModel):
    name = CharField(null=True, unique=True)
    builtin = BooleanField(default = False)
    pseudo = BooleanField(default = False)

@table
class Tag(BaseModel):
    name = CharField()
    prefix = ForeignKeyField(Prefix, null=True, related_name = 'tag')

@table
class DocumentTag(BaseModel):
    document = ForeignKeyField(Document, related_name = 'tags')
    tag = ForeignKeyField(Tag)

    class Meta:
        primary_key = CompositeKey('document', 'tag')

@table
class TagImplications(BaseModel):
    tag = ForeignKeyField(Tag)
    implies_tag = ForeignKeyField(Tag, related_name = 'implications')
    
    class Meta:
        primary_key = CompositeKey('tag', 'implies_tag')