Add db & pydantic models + set up alembic

2025-06-02 08:18:12 +02:00
parent fa0530e18c
commit aa64491313
33 changed files with 648 additions and 16 deletions
@@ -0,0 +1,90 @@
+"""Rebuild
+
+Revision ID: 5840d2b52dd8
+Revises: 
+Create Date: 2025-06-01 14:27:25.657473
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+import sqlmodel
+
+
+# revision identifiers, used by Alembic.
+revision: str = '5840d2b52dd8'
+down_revision: Union[str, None] = None
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('client',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('tin_number', sa.Integer(), nullable=False),
+    sa.Column('names', sqlmodel.sql.sqltypes.AutoString(length=100), nullable=False),
+    sa.Column('phone_number', sqlmodel.sql.sqltypes.AutoString(length=10), nullable=False),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('tin_number')
+    )
+    op.create_table('product',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('product_code', sqlmodel.sql.sqltypes.AutoString(length=10), nullable=False),
+    sa.Column('product_name', sqlmodel.sql.sqltypes.AutoString(length=20), nullable=False),
+    sa.Column('purchase_price', sa.Integer(), nullable=False),
+    sa.Column('date_modified', sa.DateTime(), server_default=sa.text('now()'), nullable=True),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('product_code'),
+    sa.UniqueConstraint('product_name')
+    )
+    op.create_table('supplier',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('tin_number', sa.Integer(), nullable=False),
+    sa.Column('names', sqlmodel.sql.sqltypes.AutoString(length=100), nullable=False),
+    sa.Column('phone_number', sqlmodel.sql.sqltypes.AutoString(length=10), nullable=False),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('tin_number')
+    )
+    op.create_table('credit',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('transcation_type', sa.Enum('BUY', 'SELL', name='tradetype'), nullable=False),
+    sa.Column('product_code', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+    sa.Column('client_id', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+    sa.Column('supplier_id', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+    sa.Column('qty', sa.Integer(), nullable=False),
+    sa.Column('amount', sa.Integer(), nullable=False),
+    sa.Column('date', sa.DateTime(), server_default=sa.text('now()'), nullable=True),
+    sa.ForeignKeyConstraint(['client_id'], ['client.id'], ),
+    sa.ForeignKeyConstraint(['product_code'], ['product.product_code'], ),
+    sa.ForeignKeyConstraint(['supplier_id'], ['supplier.id'], ),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_table('payment',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('payment_type', sa.Enum('BUY', 'SELL', name='tradetype'), nullable=False),
+    sa.Column('product_code', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+    sa.Column('client_id', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+    sa.Column('supplier_id', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+    sa.Column('amount', sa.Integer(), nullable=False),
+    sa.Column('payment_method', sqlmodel.sql.sqltypes.AutoString(length=24), nullable=False),
+    sa.Column('date', sa.DateTime(), server_default=sa.text('now()'), nullable=True),
+    sa.ForeignKeyConstraint(['client_id'], ['client.id'], ),
+    sa.ForeignKeyConstraint(['product_code'], ['product.product_code'], ),
+    sa.ForeignKeyConstraint(['supplier_id'], ['supplier.id'], ),
+    sa.PrimaryKeyConstraint('id')
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('payment')
+    op.drop_table('credit')
+    op.drop_table('supplier')
+    op.drop_table('product')
+    op.drop_table('client')
+    # ### end Alembic commands ###
@@ -0,0 +1,60 @@
+"""Fix client_id in Credit type to int
+
+Revision ID: bfb086d8d500
+Revises: 5840d2b52dd8
+Create Date: 2025-06-01 14:53:57.095181
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+import sqlmodel
+
+# revision identifiers, used by Alembic.
+revision: str = 'bfb086d8d500'
+down_revision: Union[str, None] = '5840d2b52dd8'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('credit',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('transcation_type', sa.Enum('BUY', 'SELL', name='tradetype'), nullable=False),
+    sa.Column('product_code', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+    sa.Column('client_id', sa.Integer(), nullable=False),
+    sa.Column('supplier_id', sa.Integer(), nullable=False),
+    sa.Column('qty', sa.Integer(), nullable=False),
+    sa.Column('amount', sa.Integer(), nullable=False),
+    sa.Column('date', sa.DateTime(), server_default=sa.text('now()'), nullable=True),
+    sa.ForeignKeyConstraint(['client_id'], ['client.id'], ),
+    sa.ForeignKeyConstraint(['product_code'], ['product.product_code'], ),
+    sa.ForeignKeyConstraint(['supplier_id'], ['supplier.id'], ),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_table('payment',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('payment_type', sa.Enum('BUY', 'SELL', name='tradetype'), nullable=False),
+    sa.Column('product_code', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+    sa.Column('client_id', sa.Integer(), nullable=False),
+    sa.Column('supplier_id', sa.Integer(), nullable=False),
+    sa.Column('amount', sa.Integer(), nullable=False),
+    sa.Column('payment_method', sqlmodel.sql.sqltypes.AutoString(length=24), nullable=False),
+    sa.Column('date', sa.DateTime(), server_default=sa.text('now()'), nullable=True),
+    sa.ForeignKeyConstraint(['client_id'], ['client.id'], ),
+    sa.ForeignKeyConstraint(['product_code'], ['product.product_code'], ),
+    sa.ForeignKeyConstraint(['supplier_id'], ['supplier.id'], ),
+    sa.PrimaryKeyConstraint('id')
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('payment')
+    op.drop_table('credit')
+    # ### end Alembic commands ###