Modularized models.py

This commit is contained in:
Candifloss 2025-03-03 16:04:59 +05:30
parent 3731a68d3c
commit 783c25d0ad

View File

@ -1,47 +1,54 @@
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy import Enum, Integer, Float, String, Date, Column
from sqlalchemy import Enum, Integer, Float, String, Date, Column, Boolean
from config import item_attributes, sql_conf
from definitions.attribute import textAttribute, intAttribute, floatAttribute, dateAttribute, selectAttribute
# Initialize SQLAlchemy
db = SQLAlchemy()
# Mapping of attribute types to SQLAlchemy column types
COLUMN_TYPE_MAPPING = {
textAttribute: lambda attrib: String(attrib.max_length),
dateAttribute: lambda attrib: Date,
selectAttribute: lambda attrib: Enum(*attrib.options),
intAttribute: lambda attrib: Integer,
floatAttribute: lambda attrib: Float,
}
def _get_column_type(attrib):
"""Helper function to get the SQLAlchemy column type for an attribute."""
if type(attrib) not in COLUMN_TYPE_MAPPING:
raise ValueError(f"Unsupported attribute type: {type(attrib)} for attribute '{attrib.attrib_name}'")
return COLUMN_TYPE_MAPPING[type(attrib)](attrib)
def _get_column_properties(attrib):
"""Helper function to get the properties for a column."""
return {
'primary_key': attrib.primary,
'unique': attrib.unique,
'nullable': not attrib.required,
'default': attrib.default_val
}
def create_asset_model():
"""
Dynamically creates the Asset model based on the configuration in item_attributes.
"""
# Define the table name and basic methods
attrs = {
'__tablename__': sql_conf.SQL_TABLE, # Table name from config
'__init__': lambda self, **kwargs: self.__dict__.update(kwargs), # Constructor
'__repr__': lambda self: f"<Asset {getattr(self, next(attrib.attrib_name for attrib in item_attributes if attrib.primary))}>" # Representation
'__tablename__': sql_conf.SQL_TABLE,
'__init__': lambda self, **kwargs: self.__dict__.update(kwargs),
'__repr__': lambda self: f"<Asset {getattr(self, next(attrib.attrib_name for attrib in item_attributes if attrib.primary))}>"
}
# Define columns based on item_attributes
for attrib in item_attributes:
# Determine the column type
if isinstance(attrib, textAttribute):
column_type = String(attrib.max_length)
elif isinstance(attrib, dateAttribute):
column_type = Date
elif isinstance(attrib, selectAttribute):
column_type = Enum(*attrib.options)
elif isinstance(attrib, (intAttribute, floatAttribute)):
column_type = Integer if isinstance(attrib, intAttribute) else Float
else:
raise ValueError(f"Unsupported attribute type: {type(attrib)}")
# Define the column with additional properties
attrs[attrib.attrib_name] = Column(
column_type,
primary_key=attrib.primary,
unique=attrib.unique,
nullable=not attrib.required,
default=attrib.default_val
)
column_type = _get_column_type(attrib)
column_properties = _get_column_properties(attrib)
attrs[attrib.attrib_name] = Column(column_type, **column_properties)
# Create the Item class dynamically
Item = type('Asset', (db.Model,), attrs)
Item = type('Item', (db.Model,), attrs)
return Item
# Create the Item model