diff --git a/alembic.ini b/alembic.ini new file mode 100644 index 0000000000000000000000000000000000000000..d7ed35596fc3ccae8010534c1ce22701facc7d29 --- /dev/null +++ b/alembic.ini @@ -0,0 +1,45 @@ +# A generic, single database configuration. + +[alembic] +# template used to generate migration files +file_template = %%(year)d-%%(month).2d-%%(day).2d_%%(rev)s_%%(slug)s + +# set to 'true' to run the environment during +# the 'revision' command, regardless of autogenerate +# revision_environment = false +script_location = migrations +version_locations = %(here)s/migrations/versions/schema +# Logging configuration +[loggers] +keys = root,sqlalchemy,alembic + +[handlers] +keys = console + +[formatters] +keys = generic + +[logger_root] +level = WARN +handlers = console +qualname = + +[logger_sqlalchemy] +level = WARN +handlers = +qualname = sqlalchemy.engine + +[logger_alembic] +level = INFO +handlers = +qualname = alembic + +[handler_console] +class = StreamHandler +args = (sys.stderr,) +level = NOTSET +formatter = generic + +[formatter_generic] +format = %(levelname)-5.5s [%(name)s] %(message)s +datefmt = %H:%M:%S \ No newline at end of file diff --git a/migrations/__pycache__/env.cpython-39.pyc b/migrations/__pycache__/env.cpython-39.pyc new file mode 100644 index 0000000000000000000000000000000000000000..a3f1f38bec2618de40ae47f038c6ae77f2e29148 Binary files /dev/null and b/migrations/__pycache__/env.cpython-39.pyc differ diff --git a/migrations/env.py b/migrations/env.py new file mode 100644 index 0000000000000000000000000000000000000000..1b039d7fe43ce29e53e06791acf246d39c1ad88e --- /dev/null +++ b/migrations/env.py @@ -0,0 +1,92 @@ +import logging +import os +from alembic import context +from sqlalchemy import engine_from_config, pool + +import orchestrator +from orchestrator.db.database import BaseModel +from orchestrator.settings import app_settings + +# this is the Alembic Config object, which provides +# access to the values within the .ini file in use. +config = context.config + +# Interpret the config file for Python logging. +# This line sets up loggers basically. +logger = logging.getLogger("alembic.env") + +config.set_main_option("sqlalchemy.url", app_settings.DATABASE_URI) + +# add your model's MetaData object here +# for 'autogenerate' support +# from myapp import mymodel +# target_metadata = mymodel.Base.metadata +target_metadata = BaseModel.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. + + """ + + # this callback is used to prevent an auto-migration from being generated + # when there are no changes to the schema + # reference: http://alembic.zzzcomputing.com/en/latest/cookbook.html + def process_revision_directives(context, revision, directives): # type: ignore + if getattr(config.cmd_opts, "autogenerate", False): + script = directives[0] + if script.upgrade_ops.is_empty(): + directives[:] = [] + logger.info("No changes in schema detected.") + + engine = engine_from_config( + config.get_section(config.config_ini_section), prefix="sqlalchemy.", poolclass=pool.NullPool + ) + + connection = engine.connect() + context.configure( + connection=connection, + target_metadata=target_metadata, + process_revision_directives=process_revision_directives, + compare_type=True, + ) + try: + with context.begin_transaction(): + context.run_migrations() + finally: + connection.close() + + +if context.is_offline_mode(): + run_migrations_offline() +else: + run_migrations_online() diff --git a/migrations/helpers.py b/migrations/helpers.py new file mode 100644 index 0000000000000000000000000000000000000000..c563452661eb495c4ef96d02869714fa4c4e107a --- /dev/null +++ b/migrations/helpers.py @@ -0,0 +1,3 @@ +from orchestrator.migrations.helpers import * + +# Write your own helper functions below this line. diff --git a/migrations/script.py.mako b/migrations/script.py.mako new file mode 100644 index 0000000000000000000000000000000000000000..e1fbdb8f5bae117c3ea2041e7d318b09f312616e --- /dev/null +++ b/migrations/script.py.mako @@ -0,0 +1,23 @@ +"""${message}. + +Revision ID: ${up_revision} +Revises: ${down_revision | comma,n} +Create Date: ${create_date} + +""" +import sqlalchemy as sa +from alembic import op +${imports if imports else ""} +# revision identifiers, used by Alembic. +revision = ${repr(up_revision)} +down_revision = ${repr(down_revision)} +branch_labels = ${repr(branch_labels)} +depends_on = ${repr(depends_on)} + + +def upgrade() -> None: + ${upgrades if upgrades else "pass"} + + +def downgrade() -> None: + ${downgrades if downgrades else "pass"} diff --git a/products/__init__.py b/products/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/products/product_blocks/trunk.py b/products/product_blocks/trunk.py new file mode 100644 index 0000000000000000000000000000000000000000..dcd27f16876049ec9b2fe03b83adc8d938adf95e --- /dev/null +++ b/products/product_blocks/trunk.py @@ -0,0 +1,22 @@ +from typing import Optional + +from orchestrator.domain.base import ProductBlockModel +from orchestrator.types import SubscriptionLifecycle + + +class TrunkBlockInactive(ProductBlockModel, lifecycle=[SubscriptionLifecycle.INITIAL], product_block_name="TrunkBlock"): + trunk_id: Optional[int] = None + geant_s_sid: Optional[str] = None + name: Optional[str] = None + + +class TrunkBlockProvisioning(TrunkBlockInactive, lifecycle=[SubscriptionLifecycle.PROVISIONING]): + trunk_id: Optional[int] = None + geant_s_sid: Optional[str] = None + name: str + + +class TrunkBlock(TrunkBlockProvisioning, lifecycle=[SubscriptionLifecycle.ACTIVE]): + trunk_id: int + geant_s_sid: str + name: str diff --git a/products/product_blocks/trunk_config.py b/products/product_blocks/trunk_config.py new file mode 100644 index 0000000000000000000000000000000000000000..bf375d0cf63c63e24d0aa5d7d1d5582204e9b42c --- /dev/null +++ b/products/product_blocks/trunk_config.py @@ -0,0 +1,21 @@ +from typing import Optional + +from orchestrator.domain.base import ProductBlockModel +from orchestrator.types import SubscriptionLifecycle + +from products.product_blocks.trunk import TrunkBlock, TrunkBlockInactive, TrunkBlockProvisioning + + +class TrunkConfigBlockInactive(ProductBlockModel, lifecycle=[SubscriptionLifecycle.INITIAL], product_block_name="TrunkConfigBlock"): + trunk: TrunkBlockInactive + trunk_config_id: Optional[int] = None + + +class TrunkConfigBlockProvisioning(TrunkConfigBlockInactive, lifecycle=[SubscriptionLifecycle.PROVISIONING]): + trunk: TrunkBlockProvisioning + trunk_config_id: Optional[int] = None + + +class TrunkConfigBlock(TrunkConfigBlockProvisioning, lifecycle=[SubscriptionLifecycle.ACTIVE]): + trunk: TrunkBlock + trunk_config_id: int diff --git a/products/product_blocks/trunk_config_common.py b/products/product_blocks/trunk_config_common.py new file mode 100644 index 0000000000000000000000000000000000000000..c2c18b985bf66078b27b91350e4e586b0a193ed7 --- /dev/null +++ b/products/product_blocks/trunk_config_common.py @@ -0,0 +1,33 @@ +from typing import Optional + +from orchestrator.domain.base import ProductBlockModel +from orchestrator.types import SubscriptionLifecycle + +from products.product_blocks.trunk_config import TrunkConfigBlock, TrunkConfigBlockInactive, TrunkConfigBlockProvisioning + + +class TrunkConfigCommonBlockInactive(ProductBlockModel, lifecycle=[SubscriptionLifecycle.INITIAL], product_block_name="TrunkConfigCommonBlock"): + trunk_config: TrunkConfigBlockInactive + trunk_config_common_id: Optional[int] = None + speed: Optional[str] = None + is_leased_line: Optional[bool] = None + isis_metric: Optional[int] = None + minimum_links: Optional[int] = None + + +class TrunkConfigCommonBlockProvisioning(TrunkConfigCommonBlockInactive, lifecycle=[SubscriptionLifecycle.PROVISIONING]): + trunk_config: TrunkConfigBlockProvisioning + trunk_config_common_id: Optional[int] = None + speed: str + is_leased_line: Optional[bool] = None + isis_metric: int + minimum_links: int + + +class TrunkConfigCommonBlock(TrunkConfigCommonBlockProvisioning, lifecycle=[SubscriptionLifecycle.ACTIVE]): + trunk_config: TrunkConfigBlock + trunk_config_common_id: int + speed: str + is_leased_line: bool + isis_metric: int + minimum_links: int diff --git a/products/product_blocks/trunk_config_side.py b/products/product_blocks/trunk_config_side.py new file mode 100644 index 0000000000000000000000000000000000000000..de792d4922ed7be7e62d4dcb9ff059863e008640 --- /dev/null +++ b/products/product_blocks/trunk_config_side.py @@ -0,0 +1,39 @@ +from typing import Optional + +from orchestrator.domain.base import ProductBlockModel +from orchestrator.types import SubscriptionLifecycle + +from products.product_blocks.trunk_config import TrunkConfigBlock, TrunkConfigBlockInactive, TrunkConfigBlockProvisioning + + +class TrunkConfigSideBlockInactive(ProductBlockModel, lifecycle=[SubscriptionLifecycle.INITIAL], product_block_name="TrunkConfigSideBlock"): + trunk_config: TrunkConfigBlockInactive + trunk_config_side_id: Optional[int] = None + name: Optional[str] = None + ae_name: Optional[str] = None + geant_a_sid: Optional[str] = None + ipv4_address: Optional[str] = None + ipv6_address: Optional[str] = None + members: Optional[list] = None + + +class TrunkConfigSideBlockProvisioning(TrunkConfigSideBlockInactive, lifecycle=[SubscriptionLifecycle.PROVISIONING]): + trunk_config: TrunkConfigBlockProvisioning + trunk_config_side_id: Optional[int] = None + name: str + ae_name: Optional[str] = None + geant_a_sid: Optional[str] = None + ipv4_address: Optional[str] = None + ipv6_address: Optional[str] = None + members: Optional[list] = None + + +class TrunkConfigSideBlock(TrunkConfigSideBlockProvisioning, lifecycle=[SubscriptionLifecycle.ACTIVE]): + trunk_config: TrunkConfigBlock + trunk_config_side_id: int + name: str + ae_name: str + geant_a_sid: str + ipv4_address: str + ipv6_address: str + members: list diff --git a/products/product_types/trunk.py b/products/product_types/trunk.py new file mode 100644 index 0000000000000000000000000000000000000000..5b000584ee2c84b90a7f37024f6ec0c71dc4b272 --- /dev/null +++ b/products/product_types/trunk.py @@ -0,0 +1,16 @@ +from orchestrator.domain.base import SubscriptionModel +from orchestrator.types import SubscriptionLifecycle + +from products.product_blocks.trunk import TrunkBlock, TrunkBlockInactive, TrunkBlockProvisioning + + +class TrunkInactive(SubscriptionModel, is_base=True, lifecycle=[SubscriptionLifecycle.INITIAL]): + trunk: TrunkBlockInactive + + +class TrunkProvisioning(TrunkInactive, lifecycle=[SubscriptionLifecycle.PROVISIONING]): + trunk: TrunkBlockProvisioning + + +class Trunk(TrunkProvisioning, lifecycle=[SubscriptionLifecycle.ACTIVE]): + trunk: TrunkBlock diff --git a/products/product_types/trunk_config.py b/products/product_types/trunk_config.py new file mode 100644 index 0000000000000000000000000000000000000000..bdd5ca7ad2a3ea943c2ece2dab11c54dc7df4c14 --- /dev/null +++ b/products/product_types/trunk_config.py @@ -0,0 +1,16 @@ +from orchestrator.domain.base import SubscriptionModel +from orchestrator.types import SubscriptionLifecycle + +from products.product_blocks.trunk_config import TrunkConfigBlock, TrunkConfigBlockInactive, TrunkConfigBlockProvisioning + + +class TrunkInactive(SubscriptionModel, is_base=True, lifecycle=[SubscriptionLifecycle.INITIAL]): + trunk_config: TrunkConfigBlockInactive + + +class TrunkProvisioning(TrunkInactive, lifecycle=[SubscriptionLifecycle.PROVISIONING]): + trunk_config: TrunkConfigBlockProvisioning + + +class Trunk(TrunkProvisioning, lifecycle=[SubscriptionLifecycle.ACTIVE]): + trunk_config: TrunkConfigBlock diff --git a/products/product_types/trunk_config_common.py b/products/product_types/trunk_config_common.py new file mode 100644 index 0000000000000000000000000000000000000000..b098f499123565feab493e0e61bcfc6ed1d77470 --- /dev/null +++ b/products/product_types/trunk_config_common.py @@ -0,0 +1,16 @@ +from orchestrator.domain.base import SubscriptionModel +from orchestrator.types import SubscriptionLifecycle + +from products.product_blocks.trunk_config_common import TrunkConfigCommonBlock, TrunkConfigCommonBlockInactive, TrunkConfigCommonBlockProvisioning + + +class TrunkInactive(SubscriptionModel, is_base=True, lifecycle=[SubscriptionLifecycle.INITIAL]): + trunk_config_common: TrunkConfigCommonBlockInactive + + +class TrunkProvisioning(TrunkInactive, lifecycle=[SubscriptionLifecycle.PROVISIONING]): + trunk_config_common: TrunkConfigCommonBlockProvisioning + + +class Trunk(TrunkProvisioning, lifecycle=[SubscriptionLifecycle.ACTIVE]): + trunk_config_common: TrunkConfigCommonBlock diff --git a/products/product_types/trunk_config_side.py b/products/product_types/trunk_config_side.py new file mode 100644 index 0000000000000000000000000000000000000000..55b057170df441a772e6a12fe9899928158084e7 --- /dev/null +++ b/products/product_types/trunk_config_side.py @@ -0,0 +1,16 @@ +from orchestrator.domain.base import SubscriptionModel +from orchestrator.types import SubscriptionLifecycle + +from products.product_blocks.trunk_config_side import TrunkConfigSideBlock, TrunkConfigSideBlockInactive, TrunkConfigSideBlockProvisioning + + +class TrunkInactive(SubscriptionModel, is_base=True, lifecycle=[SubscriptionLifecycle.INITIAL]): + trunk_config_side: TrunkConfigSideBlockInactive + + +class TrunkProvisioning(TrunkInactive, lifecycle=[SubscriptionLifecycle.PROVISIONING]): + trunk_config_side: TrunkConfigSideBlockProvisioning + + +class Trunk(TrunkProvisioning, lifecycle=[SubscriptionLifecycle.ACTIVE]): + trunk_config_side: TrunkConfigSideBlock