Initial commit - Migrated to Dev environment
This commit is contained in:
35
backend/seed_models.py
Executable file
35
backend/seed_models.py
Executable file
@@ -0,0 +1,35 @@
|
||||
import asyncio
|
||||
from sqlalchemy import text
|
||||
from app.db.session import engine
|
||||
|
||||
# Discovery adatok a 200 márkához (példa bővíthető)
|
||||
MODELS_DISCOVERY = {
|
||||
"Toyota": ["Corolla", "Yaris", "Hilux", "RAV4", "C-HR", "Avensis", "Land Cruiser"],
|
||||
"Volkswagen": ["Golf", "Passat", "Polo", "Tiguan", "Touran", "Transporter", "Caddy"],
|
||||
"BMW": ["3 Series", "5 Series", "X5", "X3", "1 Series", "7 Series"],
|
||||
"Honda": ["Civic", "CR-V", "Jazz", "Accord", "CB 500", "Africa Twin"], # Autó és motor is!
|
||||
"Yamaha": ["MT-07", "R1", "Tracer 900", "Ténéré 700", "XMAX"],
|
||||
"Scania": ["R-series", "S-series", "G-series", "P-series"]
|
||||
}
|
||||
|
||||
async def discovery_bot():
|
||||
async with engine.begin() as conn:
|
||||
print("🤖 Jármű típus felderítő bot indul...")
|
||||
|
||||
# Lekérjük az összes márkát
|
||||
res = await conn.execute(text("SELECT id, name FROM data.vehicle_brands"))
|
||||
brands = res.fetchall()
|
||||
|
||||
count = 0
|
||||
for b_id, b_name in brands:
|
||||
if b_name in MODELS_DISCOVERY:
|
||||
for model in MODELS_DISCOVERY[b_name]:
|
||||
await conn.execute(text(
|
||||
"INSERT INTO data.vehicle_models (brand_id, name) VALUES (:b, :n) ON CONFLICT DO NOTHING"
|
||||
), {"b": b_id, "n": model})
|
||||
count += 1
|
||||
|
||||
print(f"✅ Bot végzett: {count} új típus rögzítve!")
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(discovery_bot())
|
||||
Reference in New Issue
Block a user