lecrapaud 0.4.0__py3-none-any.whl

lecrapaud/config.py

@@ -0,0 +1,16 @@
+import os
+from dotenv import load_dotenv
+
+load_dotenv(override=False)
+
+PYTHON_ENV = os.getenv("PYTHON_ENV")
+REDIS_URL = os.getenv("REDIS_URL", "redis://localhost:6379")
+EMAIL = os.getenv("EMAIL")
+DATASET_ID = os.getenv("DATASET_ID")
+RECEIVER_EMAIL = os.getenv("RECEIVER_EMAIL")
+USERNAME = os.getenv("USERNAME")
+FRAISE = os.getenv("FRAISE")
+FA2 = os.getenv("2FA")
+INT = os.getenv("INT")
+LOGGING_LEVEL = os.getenv("LOGGING_LEVEL", "INFO")
+ALPHA_VENTAGE_API_KEY = os.getenv("ALPHA_VENTAGE_API_KEY")

lecrapaud/db/alembic/README

@@ -0,0 +1 @@
+Generic single-database configuration.

lecrapaud/db/alembic/env.py

@@ -0,0 +1,78 @@
+from logging.config import fileConfig
+
+from sqlalchemy import engine_from_config
+from sqlalchemy import pool
+
+from alembic import context
+from src.db.setup import DATABASE_URL
+
+# this is the Alembic Config object, which provides
+# access to the values within the .ini file in use.
+config = context.config
+config.set_main_option("sqlalchemy.url", DATABASE_URL)
+
+# Interpret the config file for Python logging.
+# This line sets up loggers basically.
+if config.config_file_name is not None:
+    fileConfig(config.config_file_name)
+
+# add your model's MetaData object here
+# for 'autogenerate' support
+from src.db.models.base import Base
+
+target_metadata = Base.metadata
+
+# other values from the config, defined by the needs of env.py,
+# can be acquired:
+# my_important_option = config.get_main_option("my_important_option")
+# ... etc.
+
+
+def run_migrations_offline() -> None:
+    """Run migrations in 'offline' mode.
+
+    This configures the context with just a URL
+    and not an Engine, though an Engine is acceptable
+    here as well.  By skipping the Engine creation
+    we don't even need a DBAPI to be available.
+
+    Calls to context.execute() here emit the given string to the
+    script output.
+
+    """
+    url = config.get_main_option("sqlalchemy.url")
+    context.configure(
+        url=url,
+        target_metadata=target_metadata,
+        literal_binds=True,
+        dialect_opts={"paramstyle": "named"},
+    )
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+def run_migrations_online() -> None:
+    """Run migrations in 'online' mode.
+
+    In this scenario we need to create an Engine
+    and associate a connection with the context.
+
+    """
+    connectable = engine_from_config(
+        config.get_section(config.config_ini_section, {}),
+        prefix="sqlalchemy.",
+        poolclass=pool.NullPool,
+    )
+
+    with connectable.connect() as connection:
+        context.configure(connection=connection, target_metadata=target_metadata)
+
+        with context.begin_transaction():
+            context.run_migrations()
+
+
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()

lecrapaud/db/alembic/script.py.mako

@@ -0,0 +1,26 @@
+"""${message}
+
+Revision ID: ${up_revision}
+Revises: ${down_revision | comma,n}
+Create Date: ${create_date}
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+${imports if imports else ""}
+
+# revision identifiers, used by Alembic.
+revision: str = ${repr(up_revision)}
+down_revision: Union[str, None] = ${repr(down_revision)}
+branch_labels: Union[str, Sequence[str], None] = ${repr(branch_labels)}
+depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
+
+
+def upgrade() -> None:
+    ${upgrades if upgrades else "pass"}
+
+
+def downgrade() -> None:
+    ${downgrades if downgrades else "pass"}

lecrapaud/db/alembic/versions/2025_04_06_1738-7390745388e4_initial_setup.py

@@ -0,0 +1,295 @@
+"""initial setup
+
+Revision ID: 7390745388e4
+Revises: 
+Create Date: 2025-04-06 17:38:29.297967
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '7390745388e4'
+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:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('datas',
+    sa.Column('id', sa.BigInteger(), autoincrement=True, nullable=False),
+    sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('updated_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('date', sa.DateTime(), nullable=True),
+    sa.Column('open', sa.Float(), nullable=True),
+    sa.Column('high', sa.Float(), nullable=True),
+    sa.Column('low', sa.Float(), nullable=True),
+    sa.Column('close', sa.Float(), nullable=True),
+    sa.Column('adjclose', sa.Float(), nullable=True),
+    sa.Column('volume', sa.BigInteger(), nullable=True),
+    sa.Column('stock', sa.String(length=50), nullable=True),
+    sa.Column('isin', sa.String(length=50), nullable=True),
+    sa.Column('security', sa.String(length=50), nullable=True),
+    sa.Column('sector', sa.String(length=255), nullable=True),
+    sa.Column('subindustry', sa.String(length=255), nullable=True),
+    sa.Column('location', sa.String(length=255), nullable=True),
+    sa.Column('vix_close', sa.Float(), nullable=True),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_datas_date'), 'datas', ['date'], unique=False)
+    op.create_index(op.f('ix_datas_id'), 'datas', ['id'], unique=False)
+    op.create_index(op.f('ix_datas_stock'), 'datas', ['stock'], unique=False)
+    op.create_table('datasets',
+    sa.Column('id', sa.BigInteger(), autoincrement=True, nullable=False),
+    sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('updated_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('name', sa.String(length=50), nullable=False),
+    sa.Column('path', sa.String(length=255), nullable=True),
+    sa.Column('type', sa.String(length=50), nullable=False),
+    sa.Column('size', sa.Integer(), nullable=False),
+    sa.Column('train_size', sa.Integer(), nullable=False),
+    sa.Column('val_size', sa.Integer(), nullable=True),
+    sa.Column('test_size', sa.Integer(), nullable=False),
+    sa.Column('number_of_stocks', sa.Integer(), nullable=False),
+    sa.Column('list_of_companies', sa.JSON(), nullable=False),
+    sa.Column('corr_threshold', sa.Float(), nullable=False),
+    sa.Column('max_features', sa.Integer(), nullable=False),
+    sa.Column('percentile', sa.Float(), nullable=False),
+    sa.Column('start_date', sa.DateTime(), nullable=False),
+    sa.Column('end_date', sa.DateTime(), nullable=False),
+    sa.Column('train_start_date', sa.DateTime(), nullable=False),
+    sa.Column('train_end_date', sa.DateTime(), nullable=False),
+    sa.Column('val_start_date', sa.DateTime(), nullable=True),
+    sa.Column('val_end_date', sa.DateTime(), nullable=True),
+    sa.Column('test_start_date', sa.DateTime(), nullable=False),
+    sa.Column('test_end_date', sa.DateTime(), nullable=False),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('name', name='uq_datasets_composite')
+    )
+    op.create_index(op.f('ix_datasets_id'), 'datasets', ['id'], unique=False)
+    op.create_table('features',
+    sa.Column('id', sa.BigInteger(), autoincrement=True, nullable=False),
+    sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('updated_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('name', sa.String(length=50), nullable=False),
+    sa.Column('type', sa.String(length=50), nullable=True),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('name')
+    )
+    op.create_index(op.f('ix_features_id'), 'features', ['id'], unique=False)
+    op.create_table('models',
+    sa.Column('id', sa.BigInteger(), autoincrement=True, nullable=False),
+    sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('updated_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('name', sa.String(length=50), nullable=False),
+    sa.Column('type', sa.String(length=50), nullable=False),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('name', 'type', name='uq_model_composite')
+    )
+    op.create_index(op.f('ix_models_id'), 'models', ['id'], unique=False)
+    op.create_table('operations',
+    sa.Column('id', sa.BigInteger(), autoincrement=True, nullable=False),
+    sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('updated_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('stock', sa.String(length=50), nullable=False),
+    sa.Column('price_in', sa.Float(), nullable=False),
+    sa.Column('date_in', sa.Date(), nullable=False),
+    sa.Column('why_buy', sa.String(length=50), nullable=True),
+    sa.Column('price_out', sa.Float(), nullable=False),
+    sa.Column('date_out', sa.Date(), nullable=False),
+    sa.Column('why_sell', sa.String(length=50), nullable=True),
+    sa.Column('pl', sa.Float(), nullable=False),
+    sa.Column('cash', sa.Float(), nullable=False),
+    sa.Column('portfolio_value', sa.Float(), nullable=False),
+    sa.Column('total_value', sa.Float(), nullable=False),
+    sa.Column('roi', sa.Float(), nullable=False),
+    sa.Column('cumulated_roi', sa.Float(), nullable=False),
+    sa.Column('cagr', sa.Float(), nullable=False),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_operations_id'), 'operations', ['id'], unique=False)
+    op.create_table('portfolios',
+    sa.Column('id', sa.BigInteger(), autoincrement=True, nullable=False),
+    sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('updated_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('stock', sa.String(length=50), nullable=False),
+    sa.Column('position', sa.Integer(), nullable=False),
+    sa.Column('price_in', sa.Float(), nullable=False),
+    sa.Column('value', sa.Float(), nullable=False),
+    sa.Column('date_in', sa.Date(), nullable=False),
+    sa.Column('why_buy', sa.String(length=50), nullable=True),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_portfolios_id'), 'portfolios', ['id'], unique=False)
+    op.create_table('targets',
+    sa.Column('id', sa.BigInteger(), autoincrement=True, nullable=False),
+    sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('updated_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('name', sa.String(length=50), nullable=False),
+    sa.Column('type', sa.String(length=50), nullable=False),
+    sa.Column('description', sa.String(length=255), nullable=True),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('name', 'type', name='uq_target_composite')
+    )
+    op.create_index(op.f('ix_targets_id'), 'targets', ['id'], unique=False)
+    op.create_table('transactions',
+    sa.Column('id', sa.BigInteger(), autoincrement=True, nullable=False),
+    sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('updated_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('buy_sell', sa.String(length=50), nullable=False),
+    sa.Column('stock', sa.String(length=50), nullable=False),
+    sa.Column('price', sa.Float(), nullable=False),
+    sa.Column('position', sa.Integer(), nullable=False),
+    sa.Column('value', sa.Float(), nullable=False),
+    sa.Column('date', sa.Date(), nullable=False),
+    sa.Column('why', sa.String(length=50), nullable=True),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_transactions_id'), 'transactions', ['id'], unique=False)
+    op.create_table('dataset_target_association',
+    sa.Column('dataset_id', sa.BigInteger(), nullable=False),
+    sa.Column('target_id', sa.BigInteger(), nullable=False),
+    sa.ForeignKeyConstraint(['dataset_id'], ['datasets.id'], ondelete='CASCADE'),
+    sa.ForeignKeyConstraint(['target_id'], ['targets.id'], ondelete='CASCADE'),
+    sa.PrimaryKeyConstraint('dataset_id', 'target_id')
+    )
+    op.create_table('feature_selections',
+    sa.Column('id', sa.BigInteger(), autoincrement=True, nullable=False),
+    sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('updated_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('training_time', sa.Integer(), nullable=True),
+    sa.Column('best_features_path', sa.String(length=255), nullable=True),
+    sa.Column('dataset_id', sa.BigInteger(), nullable=False),
+    sa.Column('target_id', sa.BigInteger(), nullable=False),
+    sa.ForeignKeyConstraint(['dataset_id'], ['datasets.id'], ondelete='CASCADE'),
+    sa.ForeignKeyConstraint(['target_id'], ['targets.id'], ondelete='CASCADE'),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('dataset_id', 'target_id', name='uq_feature_selection_composite')
+    )
+    op.create_index(op.f('ix_feature_selections_id'), 'feature_selections', ['id'], unique=False)
+    op.create_table('model_selections',
+    sa.Column('id', sa.BigInteger(), autoincrement=True, nullable=False),
+    sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('updated_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('best_model_params', sa.JSON(), nullable=True),
+    sa.Column('best_model_path', sa.String(length=255), nullable=True),
+    sa.Column('best_model_id', sa.BigInteger(), nullable=True),
+    sa.Column('target_id', sa.BigInteger(), nullable=False),
+    sa.Column('dataset_id', sa.BigInteger(), nullable=False),
+    sa.ForeignKeyConstraint(['best_model_id'], ['models.id'], ondelete='CASCADE'),
+    sa.ForeignKeyConstraint(['dataset_id'], ['datasets.id'], ondelete='CASCADE'),
+    sa.ForeignKeyConstraint(['target_id'], ['targets.id'], ondelete='CASCADE'),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('target_id', 'dataset_id', name='uq_model_selection_composite')
+    )
+    op.create_index(op.f('ix_model_selections_id'), 'model_selections', ['id'], unique=False)
+    op.create_table('feature_selection_association',
+    sa.Column('feature_selection_id', sa.BigInteger(), nullable=False),
+    sa.Column('feature_id', sa.BigInteger(), nullable=False),
+    sa.ForeignKeyConstraint(['feature_id'], ['features.id'], ondelete='CASCADE'),
+    sa.ForeignKeyConstraint(['feature_selection_id'], ['feature_selections.id'], ondelete='CASCADE'),
+    sa.PrimaryKeyConstraint('feature_selection_id', 'feature_id')
+    )
+    op.create_table('feature_selection_ranks',
+    sa.Column('id', sa.BigInteger(), autoincrement=True, nullable=False),
+    sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('updated_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('score', sa.Float(), nullable=True),
+    sa.Column('pvalue', sa.Float(), nullable=True),
+    sa.Column('support', sa.Integer(), nullable=True),
+    sa.Column('rank', sa.Integer(), nullable=True),
+    sa.Column('method', sa.String(length=50), nullable=True),
+    sa.Column('training_time', sa.Integer(), nullable=True),
+    sa.Column('feature_id', sa.BigInteger(), nullable=True),
+    sa.Column('feature_selection_id', sa.BigInteger(), nullable=True),
+    sa.ForeignKeyConstraint(['feature_id'], ['features.id'], ondelete='CASCADE'),
+    sa.ForeignKeyConstraint(['feature_selection_id'], ['feature_selections.id'], ondelete='CASCADE'),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('feature_id', 'feature_selection_id', 'method', name='uq_feature_selection_rank_composite')
+    )
+    op.create_index(op.f('ix_feature_selection_ranks_id'), 'feature_selection_ranks', ['id'], unique=False)
+    op.create_table('model_trainings',
+    sa.Column('id', sa.BigInteger(), autoincrement=True, nullable=False),
+    sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('updated_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('best_params', sa.JSON(), nullable=True),
+    sa.Column('model_path', sa.String(length=255), nullable=True),
+    sa.Column('training_time', sa.Integer(), nullable=True),
+    sa.Column('model_id', sa.BigInteger(), nullable=False),
+    sa.Column('model_selection_id', sa.BigInteger(), nullable=False),
+    sa.ForeignKeyConstraint(['model_id'], ['models.id'], ),
+    sa.ForeignKeyConstraint(['model_selection_id'], ['model_selections.id'], ondelete='CASCADE'),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('model_id', 'model_selection_id', name='uq_model_training_composite')
+    )
+    op.create_index(op.f('ix_model_trainings_id'), 'model_trainings', ['id'], unique=False)
+    op.create_table('scores',
+    sa.Column('id', sa.BigInteger(), autoincrement=True, nullable=False),
+    sa.Column('created_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('updated_at', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('type', sa.String(length=50), nullable=False),
+    sa.Column('training_time', sa.Integer(), nullable=True),
+    sa.Column('eval_data_std', sa.Float(), nullable=True),
+    sa.Column('rmse', sa.Float(), nullable=True),
+    sa.Column('rmse_std_ratio', sa.Float(), nullable=True),
+    sa.Column('mae', sa.Float(), nullable=True),
+    sa.Column('mape', sa.Float(), nullable=True),
+    sa.Column('mam', sa.Float(), nullable=True),
+    sa.Column('mad', sa.Float(), nullable=True),
+    sa.Column('mae_mam_ratio', sa.Float(), nullable=True),
+    sa.Column('mae_mad_ratio', sa.Float(), nullable=True),
+    sa.Column('r2', sa.Float(), nullable=True),
+    sa.Column('logloss', sa.Float(), nullable=True),
+    sa.Column('accuracy', sa.Float(), nullable=True),
+    sa.Column('precision', sa.Float(), nullable=True),
+    sa.Column('recall', sa.Float(), nullable=True),
+    sa.Column('f1', sa.Float(), nullable=True),
+    sa.Column('roc_auc', sa.Float(), nullable=True),
+    sa.Column('threshold', sa.Float(), nullable=True),
+    sa.Column('precision_at_threshold', sa.Float(), nullable=True),
+    sa.Column('recall_at_threshold', sa.Float(), nullable=True),
+    sa.Column('model_training_id', sa.BigInteger(), nullable=False),
+    sa.ForeignKeyConstraint(['model_training_id'], ['model_trainings.id'], ondelete='CASCADE'),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_scores_id'), 'scores', ['id'], unique=False)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_index(op.f('ix_scores_id'), table_name='scores')
+    op.drop_table('scores')
+    op.drop_index(op.f('ix_model_trainings_id'), table_name='model_trainings')
+    op.drop_table('model_trainings')
+    op.drop_index(op.f('ix_feature_selection_ranks_id'), table_name='feature_selection_ranks')
+    op.drop_table('feature_selection_ranks')
+    op.drop_table('feature_selection_association')
+    op.drop_index(op.f('ix_model_selections_id'), table_name='model_selections')
+    op.drop_table('model_selections')
+    op.drop_index(op.f('ix_feature_selections_id'), table_name='feature_selections')
+    op.drop_table('feature_selections')
+    op.drop_table('dataset_target_association')
+    op.drop_index(op.f('ix_transactions_id'), table_name='transactions')
+    op.drop_table('transactions')
+    op.drop_index(op.f('ix_targets_id'), table_name='targets')
+    op.drop_table('targets')
+    op.drop_index(op.f('ix_portfolios_id'), table_name='portfolios')
+    op.drop_table('portfolios')
+    op.drop_index(op.f('ix_operations_id'), table_name='operations')
+    op.drop_table('operations')
+    op.drop_index(op.f('ix_models_id'), table_name='models')
+    op.drop_table('models')
+    op.drop_index(op.f('ix_features_id'), table_name='features')
+    op.drop_table('features')
+    op.drop_index(op.f('ix_datasets_id'), table_name='datasets')
+    op.drop_table('datasets')
+    op.drop_index(op.f('ix_datas_stock'), table_name='datas')
+    op.drop_index(op.f('ix_datas_id'), table_name='datas')
+    op.drop_index(op.f('ix_datas_date'), table_name='datas')
+    op.drop_table('datas')
+    # ### end Alembic commands ###

lecrapaud/db/alembic/versions/2025_04_06_1755-40cd8d3e798e_unique_constraint_for_data.py

@@ -0,0 +1,30 @@
+"""unique constraint for data
+
+Revision ID: 40cd8d3e798e
+Revises: 7390745388e4
+Create Date: 2025-04-06 17:55:47.709956
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '40cd8d3e798e'
+down_revision: Union[str, None] = '7390745388e4'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_unique_constraint('uq_datas_composite', 'datas', ['date', 'stock'])
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint('uq_datas_composite', 'datas', type_='unique')
+    # ### end Alembic commands ###

lecrapaud/db/alembic/versions/2025_05_23_1724-2360941fa0bd_longer_string.py

@@ -0,0 +1,52 @@
+"""longer_string
+
+Revision ID: 2360941fa0bd
+Revises: 40cd8d3e798e
+Create Date: 2025-05-23 17:24:07.750472
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import mysql
+
+# revision identifiers, used by Alembic.
+revision: str = '2360941fa0bd'
+down_revision: Union[str, None] = '40cd8d3e798e'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('operations', 'why_buy',
+               existing_type=mysql.VARCHAR(length=50),
+               type_=sa.String(length=255),
+               existing_nullable=True)
+    op.alter_column('operations', 'why_sell',
+               existing_type=mysql.VARCHAR(length=50),
+               type_=sa.String(length=255),
+               existing_nullable=True)
+    op.alter_column('transactions', 'why',
+               existing_type=mysql.VARCHAR(length=50),
+               type_=sa.String(length=255),
+               existing_nullable=True)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('transactions', 'why',
+               existing_type=sa.String(length=255),
+               type_=mysql.VARCHAR(length=50),
+               existing_nullable=True)
+    op.alter_column('operations', 'why_sell',
+               existing_type=sa.String(length=255),
+               type_=mysql.VARCHAR(length=50),
+               existing_nullable=True)
+    op.alter_column('operations', 'why_buy',
+               existing_type=sa.String(length=255),
+               type_=mysql.VARCHAR(length=50),
+               existing_nullable=True)
+    # ### end Alembic commands ###

lecrapaud/db/alembic/versions/2025_05_27_1159-b96396dcfaff_add_env_to_trading_tables.py

@@ -0,0 +1,34 @@
+"""add_env_to_trading_tables
+
+Revision ID: b96396dcfaff
+Revises: 2360941fa0bd
+Create Date: 2025-05-27 11:59:55.405410
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'b96396dcfaff'
+down_revision: Union[str, None] = '2360941fa0bd'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('operations', sa.Column('env', sa.String(length=50), nullable=False))
+    op.add_column('portfolios', sa.Column('env', sa.String(length=50), nullable=False))
+    op.add_column('transactions', sa.Column('env', sa.String(length=50), nullable=False))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('transactions', 'env')
+    op.drop_column('portfolios', 'env')
+    op.drop_column('operations', 'env')
+    # ### end Alembic commands ###

lecrapaud/db/alembic/versions/2025_05_27_1337-40cbfc215f7c_fix_nb_character_on_portfolio.py

@@ -0,0 +1,39 @@
+"""fix_nb_character_on_portfolio
+
+
+Revision ID: 40cbfc215f7c
+Revises: b96396dcfaff
+Create Date: 2025-05-27 13:37:18.291145
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import mysql
+
+# revision identifiers, used by Alembic.
+revision: str = '40cbfc215f7c'
+down_revision: Union[str, None] = 'b96396dcfaff'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('portfolios', 'why_buy',
+               existing_type=mysql.VARCHAR(length=50),
+               type_=sa.String(length=255),
+               existing_nullable=True)
+    op.create_unique_constraint('uq_portfolio_composite', 'portfolios', ['stock', 'env'])
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint('uq_portfolio_composite', 'portfolios', type_='unique')
+    op.alter_column('portfolios', 'why_buy',
+               existing_type=sa.String(length=255),
+               type_=mysql.VARCHAR(length=50),
+               existing_nullable=True)
+    # ### end Alembic commands ###

lecrapaud/db/alembic/versions/2025_05_27_1526-3de994115317_to_datetime.py

@@ -0,0 +1,36 @@
+"""to_datetime
+
+Revision ID: 3de994115317
+Revises: 40cbfc215f7c
+Create Date: 2025-05-27 15:26:01.669012
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '3de994115317'
+down_revision: Union[str, None] = '40cbfc215f7c'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('transactions', 'date',
+               existing_type=sa.DATE(),
+               type_=sa.DateTime(),
+               existing_nullable=False)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('transactions', 'date',
+               existing_type=sa.DateTime(),
+               type_=sa.DATE(),
+               existing_nullable=False)
+    # ### end Alembic commands ###

lecrapaud/db/alembic/versions/2025_05_27_2003-25c227c684f8_add_fees_to_transactions.py

@@ -0,0 +1,30 @@
+"""add_fees_to_transactions
+
+Revision ID: 25c227c684f8
+Revises: 3de994115317
+Create Date: 2025-05-27 20:03:08.193307
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '25c227c684f8'
+down_revision: Union[str, None] = '3de994115317'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('transactions', sa.Column('fees', sa.Float(), nullable=False))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('transactions', 'fees')
+    # ### end Alembic commands ###