summaryrefslogtreecommitdiff
path: root/archivist/model.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--archivist/model.py165
1 files changed, 59 insertions, 106 deletions
diff --git a/archivist/model.py b/archivist/model.py
index 69765f3..b281edc 100644
--- a/archivist/model.py
+++ b/archivist/model.py
@@ -1,106 +1,59 @@
-from sqlalchemy import create_engine
-from sqlalchemy import MetaData, Table, Column, ForeignKey, UniqueConstraint
-from sqlalchemy import types as ty
-
-from sqlalchemy.orm import relationship
-from sqlalchemy.ext.declarative import as_declarative, declared_attr
-
-from functools import partial
-
-from . import Session
-
-engine = create_engine('sqlite:///:memory:', echo=True)
-Session.configure(bind=engine)
-
-ReqColumn = partial(Column, nullable = False)
-
-convention = {
- 'ix': "ix_%(column_0_label)s",
- 'uq': "uq_%(table_name)s_%(column_0_name)s",
- 'ck': "ck_%(table_name)s_%(column_0_name)s",
- 'fk': "fk_%(table_name)s_%(column_0_name)s_%(referred_table_name)s",
- 'pk': "pk_%(table_name)s"
-}
-metadata = MetaData(naming_convention = convention)
-
-try:
- from sqlalchemy_repr import RepresentableBase as Base
-except ImportError:
- Base = object
-
-class _QueryProperty(object):
- def __get__(self, obj, type):
- return Session().query(type)
-
-@as_declarative(metadata = metadata)
-class Model(Base):
- id = Column(ty.Integer, primary_key=True)
-
- @declared_attr
- def __tablename__ (cls):
- return cls.__name__.lower()
-
- query = _QueryProperty()
-
- @classmethod
- def get_by(cls, *args, **kwargs):
- return cls.query.filter_by(*args, **kwargs).first()
-
- @classmethod
- def get(cls, *args, **kwargs):
- return cls.query.get(*args, **kwargs)
-
-def create_all():
- Model.metadata.create_all(engine)
-
-def drop_all():
- Model.metadata.drop_all(engine)
-
-class Prefix(Model):
- prefix = Column(ty.Unicode, index = True, unique = True)
- builtin = ReqColumn(ty.Boolean, default = False)
- pseudo = ReqColumn(ty.Boolean, default = False)
-
- def __init__(self, name):
- self.name = name
-
-class Tag(Model):
- name = ReqColumn(ty.Unicode)
- prefix_id = Column(ty.Integer, ForeignKey(Prefix.id))
- prefix = relationship('Prefix', backref='tag')
-
- implications = relationship('Tag',
- secondary = 'tag_x_tag',
- primaryjoin = 'Tag.id == tag_x_tag.c.tag_id',
- secondaryjoin = 'Tag.id == tag_x_tag.c.implies_tag_id',
- backref='implied_by')
-
- def __init__(self, name, prefix = None):
- self.name = name
- self.prefix = prefix
-
- def implies(self, other):
- self.implications.append(other)
-
- def remove_implication(self, other):
- self.implications.remove(other)
-
-class Document(Model):
- description = Column(ty.Unicode)
- created = ReqColumn(ty.DateTime)
- content = ReqColumn(ty.LargeBinary)
- original_path = Column(ty.Unicode)
- direction = Column(ty.Boolean)
- tags = relationship('Tag', secondary = 'document_x_tag')
-
-Table('document_x_tag', Model.metadata,
- Column('document_id', ty.Integer, ForeignKey(Document.id)),
- Column('tag_id', ty.Integer, ForeignKey(Tag.id)),
- UniqueConstraint('document_id', 'tag_id', name = 'uq_document_x_tag')
- )
-
-Table('tag_x_tag', Model.metadata,
- Column('tag_id', ty.Integer, ForeignKey(Tag.id)),
- Column('implies_tag_id', ty.Integer, ForeignKey(Tag.id)),
- UniqueConstraint('tag_id', 'implies_tag_id', name = 'uq_tag_x_tag')
- )
+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')