143 lines
4.6 KiB
Python
143 lines
4.6 KiB
Python
"""Celery tasks for Nexus POS background jobs."""
|
|
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
# Ensure pos/ is on path for imports when Celery worker runs standalone
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
from celery_app import celery
|
|
import psycopg2
|
|
import redis as redis_lib
|
|
|
|
MASTER_DB_URL = os.environ.get('MASTER_DB_URL', 'postgresql://postgres@/nexus_autoparts')
|
|
REDIS_URL = os.environ.get('REDIS_URL', 'redis://localhost:6379/0')
|
|
|
|
|
|
def _get_db():
|
|
return psycopg2.connect(MASTER_DB_URL)
|
|
|
|
|
|
def _get_redis():
|
|
return redis_lib.from_url(REDIS_URL, decode_responses=True)
|
|
|
|
|
|
@celery.task(bind=True, max_retries=3)
|
|
def warm_vehicle_cache_task(self, batch_size=5000, ttl=3600):
|
|
"""Warm Redis cache for vehicle info from part_vehicle_preview."""
|
|
conn = _get_db()
|
|
cur = conn.cursor()
|
|
r = _get_redis()
|
|
r.ping()
|
|
|
|
cur.execute("SELECT id_part FROM parts WHERE oem_part_number IS NOT NULL ORDER BY id_part")
|
|
all_ids = [row[0] for row in cur.fetchall()]
|
|
total = len(all_ids)
|
|
|
|
processed = 0
|
|
cached = 0
|
|
start = time.time()
|
|
|
|
for i in range(0, total, batch_size):
|
|
batch = all_ids[i:i + batch_size]
|
|
cur.execute("""
|
|
SELECT part_id, name_brand, name_model, year_car
|
|
FROM part_vehicle_preview
|
|
WHERE part_id = ANY(%s)
|
|
""", (batch,))
|
|
|
|
pipe = r.pipeline()
|
|
batch_cached = 0
|
|
for row in cur.fetchall():
|
|
info = f"{row[1]} {row[2]} {row[3]}"
|
|
pipe.setex(f'nexus:vehicle:{row[0]}', ttl, info)
|
|
batch_cached += 1
|
|
pipe.execute()
|
|
|
|
processed += len(batch)
|
|
cached += batch_cached
|
|
self.update_state(
|
|
state='PROGRESS',
|
|
meta={'current': processed, 'total': total, 'cached': cached}
|
|
)
|
|
|
|
cur.close()
|
|
conn.close()
|
|
elapsed = time.time() - start
|
|
return {'total': total, 'cached': cached, 'elapsed': int(elapsed)}
|
|
|
|
|
|
@celery.task(bind=True, max_retries=2)
|
|
def bulk_import_inventory_task(self, csv_path, tenant_id, branch_id=None):
|
|
"""Bulk import inventory from CSV in background."""
|
|
from services.inventory_engine import bulk_import_csv
|
|
conn = _get_db()
|
|
try:
|
|
result = bulk_import_csv(conn, csv_path, tenant_id, branch_id)
|
|
conn.commit()
|
|
return result
|
|
except Exception as exc:
|
|
conn.rollback()
|
|
raise self.retry(exc=exc)
|
|
finally:
|
|
conn.close()
|
|
|
|
|
|
@celery.task(bind=True, max_retries=1)
|
|
def generate_report_task(self, report_type, params, tenant_id):
|
|
"""Generate heavy reports asynchronously."""
|
|
# Placeholder: implement actual report generation per type
|
|
self.update_state(state='PROGRESS', meta={'step': 'collecting_data'})
|
|
time.sleep(2) # Simulate work
|
|
return {
|
|
'report_type': report_type,
|
|
'tenant_id': tenant_id,
|
|
'status': 'completed',
|
|
'url': f'/pos/static/reports/{report_type}_{tenant_id}.pdf',
|
|
}
|
|
|
|
|
|
@celery.task(bind=True, max_retries=2)
|
|
def sync_vehicle_compatibility_task(self, tenant_id, item_id, part_number, name, brand, compat_source):
|
|
"""Fetch AI/TecDoc vehicle compatibility in background after item creation."""
|
|
from tenant_db import get_tenant_conn, get_master_conn
|
|
from services.inventory_vehicle_compat import auto_match_vehicle_compatibility, save_qwen_fitment
|
|
from services.qwen_fitment import get_vehicle_fitment
|
|
|
|
tenant_conn = None
|
|
master_conn = None
|
|
try:
|
|
tenant_conn = get_tenant_conn(tenant_id)
|
|
|
|
if compat_source in ('tecdoc', 'both'):
|
|
try:
|
|
master_conn = get_master_conn()
|
|
auto_match_vehicle_compatibility(
|
|
master_conn, tenant_conn, item_id, part_number,
|
|
brand=brand, name=name
|
|
)
|
|
master_conn.close()
|
|
master_conn = None
|
|
except Exception as am_err:
|
|
print(f"[sync_vehicle_compat] TecDoc error for item {item_id}: {am_err}")
|
|
|
|
if compat_source in ('qwen', 'both'):
|
|
try:
|
|
fitment = get_vehicle_fitment(part_number, name, brand or '')
|
|
save_qwen_fitment(tenant_conn, item_id, fitment)
|
|
except Exception as qwen_err:
|
|
print(f"[sync_vehicle_compat] QWEN error for item {item_id}: {qwen_err}")
|
|
|
|
tenant_conn.commit()
|
|
return {'status': 'ok', 'item_id': item_id, 'tenant_id': tenant_id}
|
|
except Exception as exc:
|
|
if tenant_conn:
|
|
tenant_conn.rollback()
|
|
raise self.retry(exc=exc, countdown=10)
|
|
finally:
|
|
if tenant_conn:
|
|
tenant_conn.close()
|
|
if master_conn:
|
|
master_conn.close()
|