71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
|
import sqlalchemy as sa
|
||
|
|
from szurubooru.model.base import Base
|
||
|
|
|
||
|
|
|
||
|
|
class PoolName(Base):
|
||
|
|
__tablename__ = 'pool_name'
|
||
|
|
|
||
|
|
pool_name_id = sa.Column('pool_name_id', sa.Integer, primary_key=True)
|
||
|
|
pool_id = sa.Column(
|
||
|
|
'pool_id',
|
||
|
|
sa.Integer,
|
||
|
|
sa.ForeignKey('pool.id'),
|
||
|
|
nullable=False,
|
||
|
|
index=True)
|
||
|
|
name = sa.Column('name', sa.Unicode(128), nullable=False, unique=True)
|
||
|
|
order = sa.Column('ord', sa.Integer, nullable=False, index=True)
|
||
|
|
|
||
|
|
def __init__(self, name: str, order: int) -> None:
|
||
|
|
self.name = name
|
||
|
|
self.order = order
|
||
|
|
|
||
|
|
class Pool(Base):
|
||
|
|
__tablename__ = 'pool'
|
||
|
|
|
||
|
|
pool_id = sa.Column('id', sa.Integer, primary_key=True)
|
||
|
|
category_id = sa.Column(
|
||
|
|
'category_id',
|
||
|
|
sa.Integer,
|
||
|
|
sa.ForeignKey('pool_category.id'),
|
||
|
|
nullable=False,
|
||
|
|
index=True)
|
||
|
|
version = sa.Column('version', sa.Integer, default=1, nullable=False)
|
||
|
|
creation_time = sa.Column('creation_time', sa.DateTime, nullable=False)
|
||
|
|
last_edit_time = sa.Column('last_edit_time', sa.DateTime)
|
||
|
|
description = sa.Column('description', sa.UnicodeText, default=None)
|
||
|
|
|
||
|
|
category = sa.orm.relationship('PoolCategory', lazy='joined')
|
||
|
|
names = sa.orm.relationship(
|
||
|
|
'PoolName',
|
||
|
|
cascade='all,delete-orphan',
|
||
|
|
lazy='joined',
|
||
|
|
order_by='PoolName.order')
|
||
|
|
|
||
|
|
# post_count = sa.orm.column_property(
|
||
|
|
# sa.sql.expression.select(
|
||
|
|
# [sa.sql.expression.func.count(PostPool.post_id)])
|
||
|
|
# .where(PostPool.pool_id == pool_id)
|
||
|
|
# .correlate_except(PostPool))
|
||
|
|
# TODO
|
||
|
|
from random import randint
|
||
|
|
post_count = sa.orm.column_property(
|
||
|
|
sa.sql.expression.select([randint(1, 1000)])
|
||
|
|
.limit(1)
|
||
|
|
.as_scalar())
|
||
|
|
|
||
|
|
first_name = sa.orm.column_property(
|
||
|
|
(
|
||
|
|
sa.sql.expression.select([PoolName.name])
|
||
|
|
.where(PoolName.pool_id == pool_id)
|
||
|
|
.order_by(PoolName.order)
|
||
|
|
.limit(1)
|
||
|
|
.as_scalar()
|
||
|
|
),
|
||
|
|
deferred=True)
|
||
|
|
|
||
|
|
|
||
|
|
__mapper_args__ = {
|
||
|
|
'version_id_col': version,
|
||
|
|
'version_id_generator': False,
|
||
|
|
}
|