69 lines
1.4 KiB
Python
69 lines
1.4 KiB
Python
"""init tables
|
|
|
|
Revision ID: 9b42b5289e30
|
|
Revises:
|
|
Create Date: 2022-10-17 11:43:07.452838
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.sql import text
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = '9b42b5289e30'
|
|
down_revision = None
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
conn = op.get_bind()
|
|
|
|
init_users_query = """
|
|
CREATE TABLE users (
|
|
user_id bigint PRIMARY KEY,
|
|
chat_id bigint
|
|
);
|
|
"""
|
|
conn.execute(text(init_users_query))
|
|
|
|
init_tags_query = """
|
|
CREATE TABLE tags (
|
|
id SERIAL PRIMARY KEY,
|
|
name varchar(100)
|
|
);
|
|
"""
|
|
conn.execute(text(init_tags_query))
|
|
|
|
init_categories_query = """
|
|
CREATE TABLE categories (
|
|
id SERIAL PRIMARY KEY,
|
|
name varchar(20)
|
|
);
|
|
"""
|
|
conn.execute(text(init_categories_query))
|
|
|
|
init_projects_query = """
|
|
CREATE TABLE projects (
|
|
id SERIAL PRIMARY KEY,
|
|
name varchar(50),
|
|
description varchar(1000),
|
|
creator bigint REFERENCES users,
|
|
contributors smallint DEFAULT 1,
|
|
status bit(1) DEFAULT B'1',
|
|
tag int REFERENCES tags,
|
|
category int REFERENCES categories,
|
|
creation_date date DEFAULT CURRENT_DATE
|
|
);
|
|
"""
|
|
conn.execute(text(init_projects_query))
|
|
|
|
|
|
def downgrade() -> None:
|
|
conn = op.get_bind()
|
|
|
|
conn.execute(text(
|
|
"DROP TABLE users, tags, categories, projects CASCADE"
|
|
))
|