viktor/wrongmove
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

run alembic migrations on startup

Browse source
This commit is contained in:
Viktor Barzin 2025-06-21 21:52:51 +00:00
parent 2f8adb99e8
commit a73eea4d9e
No known key found for this signature in database
GPG key ID: 4056458DBDBF8863
2 changed files with 15 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

16
crawler/api/app.py
View file

@ -21,8 +21,22 @@ from repositories.listing_repository import ListingRepository
from database import engine from database import engine
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from ui_exporter import export_immoweb from ui_exporter import export_immoweb
from alembic import command
from alembic.config import Config
from contextlib import asynccontextmanager
app = FastAPI()
@asynccontextmanager
async def lifespan(app: FastAPI):
alembic_cfg = Config("./alembic.ini")
print("Running alembic migrations")
command.upgrade(alembic_cfg, "head")
print("Finished running alembic migrations")
yield
print("Shutting down")
app = FastAPI(lifespan=lifespan)
# Start worker thread # Start worker thread
WorkerManager(DumpListingsWorker()).start() WorkerManager(DumpListingsWorker()).start()

2
crawler/api/worker.py
View file

@ -10,8 +10,6 @@ import queue
from threading import Thread from threading import Thread
from database import engine from database import engine
from models.listing import Listing, QueryParameters
from pydantic import BaseModel
from repositories.listing_repository import ListingRepository from repositories.listing_repository import ListingRepository
dump_listings_module = importlib.import_module("1_dump_listings") dump_listings_module = importlib.import_module("1_dump_listings")