Mateusz Gruszczyński
35 lines
1.5 KiB
Python
35 lines
1.5 KiB
Python
from datetime import datetime
|
|
from .extensions import db
|
|
from .models import FuelStationCompany, FuelStationPoint
|
|
|
|
|
|
def sync_station_catalog(companies):
|
|
added = updated = 0
|
|
incoming_keys = {row["ure_key"] for row in companies}
|
|
for source in companies:
|
|
row = dict(source)
|
|
point_rows = row.pop("points", [])
|
|
with db.session.no_autoflush:
|
|
company = FuelStationCompany.query.filter_by(ure_key=row["ure_key"]).first()
|
|
if company is None:
|
|
company = FuelStationCompany(**row, active=True)
|
|
db.session.add(company); db.session.flush(); added += 1
|
|
else:
|
|
for key, value in row.items(): setattr(company, key, value)
|
|
company.active = True
|
|
company.fetched_at = datetime.utcnow(); updated += 1
|
|
existing = {p.ure_dkn: p for p in company.points.all()}
|
|
incoming = set()
|
|
for data in point_rows:
|
|
key = data["ure_dkn"]; incoming.add(key)
|
|
point = existing.get(key)
|
|
if point is None:
|
|
point = FuelStationPoint(station_company_id=company.id, ure_dkn=key); db.session.add(point)
|
|
for field, value in data.items(): setattr(point, field, value)
|
|
for key, point in existing.items():
|
|
if key not in incoming: db.session.delete(point)
|
|
company.station_count = len(incoming)
|
|
FuelStationCompany.query.filter(~FuelStationCompany.ure_key.in_(incoming_keys)).update({FuelStationCompany.active: False}, synchronize_session=False)
|
|
db.session.flush()
|
|
return added, updated
|