mirror of
https://github.com/creyD/creyPY.git
synced 2026-04-12 19:30:30 +02:00
Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 5b74ed5620 | |||
| bb3a52295d |
@@ -1,3 +1,4 @@
|
|||||||
|
from .fields import * # noqa
|
||||||
from .groups import * # noqa
|
from .groups import * # noqa
|
||||||
from .i18n import * # noqa
|
from .i18n import * # noqa
|
||||||
from .stripe import * # noqa
|
from .stripe import * # noqa
|
||||||
|
|||||||
17
creyPY/const/fields.py
Normal file
17
creyPY/const/fields.py
Normal file
@@ -0,0 +1,17 @@
|
|||||||
|
from sqlalchemy import types
|
||||||
|
|
||||||
|
|
||||||
|
class LowerCaseString(types.TypeDecorator):
|
||||||
|
"""Converts strings to lower case on the way in."""
|
||||||
|
|
||||||
|
impl = types.String
|
||||||
|
cache_ok = True
|
||||||
|
|
||||||
|
def process_bind_param(self, value, dialect):
|
||||||
|
if value is None:
|
||||||
|
return value
|
||||||
|
return value.lower()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def python_type(self):
|
||||||
|
return str
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
import uuid
|
import uuid
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
|
||||||
from sqlalchemy import Column, DateTime, String
|
from sqlalchemy import Column, DateTime, PrimaryKeyConstraint, String
|
||||||
from sqlalchemy.dialects.postgresql import UUID
|
from sqlalchemy.dialects.postgresql import UUID
|
||||||
from sqlalchemy.ext.declarative import declared_attr
|
from sqlalchemy.ext.declarative import declared_attr
|
||||||
from sqlalchemy.orm import as_declarative
|
from sqlalchemy.orm import as_declarative
|
||||||
@@ -23,6 +23,11 @@ class Base(AutoAnnotateMixin, AutoInitMixin):
|
|||||||
|
|
||||||
# TODO: Add automated foreign key resolution
|
# TODO: Add automated foreign key resolution
|
||||||
|
|
||||||
|
# Add name to primary key constraint to ensure alembic can pick it up later
|
||||||
|
@declared_attr
|
||||||
|
def __table_args__(cls):
|
||||||
|
return (PrimaryKeyConstraint("id", name=f"pk_{cls.__tablename__}"),)
|
||||||
|
|
||||||
# Generate __tablename__ automatically
|
# Generate __tablename__ automatically
|
||||||
@declared_attr
|
@declared_attr
|
||||||
def __tablename__(cls) -> str:
|
def __tablename__(cls) -> str:
|
||||||
|
|||||||
Reference in New Issue
Block a user