mirror of
https://github.com/kjanat/livegraphs-django.git
synced 2026-02-13 12:55:42 +01:00
Kaj Kowalski
2236eeb9a5
Introduces uv-based Docker workflow with non-root runtime, cached installs, and uv-run for web and Celery. Updates docker-compose to Postgres + Redis, loads .env, and removes source bind mount for reproducible builds. Switches settings to use Postgres when env is present with SQLite fallback; broadens allowed hosts for containerized development. Adds psycopg2-binary and updates sample env for Redis in Docker. Adds company scoping to external data models and links sessions during ingestion; provides management commands to seed a Jumbo company/users and sync external chat data into the dashboard. Includes .dockerignore, TypeScript config and typings, and minor template/docs tweaks. Requires database migration.
108 lines
4.4 KiB
Python
108 lines
4.4 KiB
Python
"""
|
|
Management command to set up Jumbo company, users, and link existing data.
|
|
"""
|
|
|
|
from accounts.models import Company, CustomUser
|
|
from data_integration.models import ChatSession, ExternalDataSource
|
|
from django.core.management.base import BaseCommand
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Set up Jumbo company, create users, and link existing external data"
|
|
|
|
def handle(self, *_args, **_options):
|
|
self.stdout.write("Setting up Jumbo company and data...")
|
|
|
|
# 1. Create Jumbo company
|
|
jumbo_company, created = Company.objects.get_or_create(
|
|
name="Jumbo", defaults={"description": "Jumbo Supermarkets - External API Data"}
|
|
)
|
|
if created:
|
|
self.stdout.write(self.style.SUCCESS("✓ Created Jumbo company"))
|
|
else:
|
|
self.stdout.write(" Jumbo company already exists")
|
|
|
|
# 2. Create admin user for Jumbo
|
|
admin_created = False
|
|
if not CustomUser.objects.filter(username="jumbo_admin").exists():
|
|
CustomUser.objects.create_user( # nosec B106
|
|
username="jumbo_admin",
|
|
email="admin@jumbo.nl",
|
|
password="jumbo123",
|
|
company=jumbo_company,
|
|
is_company_admin=True,
|
|
)
|
|
self.stdout.write(self.style.SUCCESS("✓ Created Jumbo admin: jumbo_admin / jumbo123"))
|
|
admin_created = True
|
|
else:
|
|
self.stdout.write(" Jumbo admin already exists")
|
|
|
|
# 3. Create regular users for Jumbo
|
|
jumbo_users = [
|
|
{
|
|
"username": "jumbo_analyst",
|
|
"email": "analyst@jumbo.nl",
|
|
"password": "jumbo123",
|
|
"is_company_admin": False,
|
|
},
|
|
{
|
|
"username": "jumbo_manager",
|
|
"email": "manager@jumbo.nl",
|
|
"password": "jumbo123",
|
|
"is_company_admin": False,
|
|
},
|
|
]
|
|
|
|
users_created = 0
|
|
for user_data in jumbo_users:
|
|
if not CustomUser.objects.filter(username=user_data["username"]).exists():
|
|
CustomUser.objects.create_user(
|
|
username=user_data["username"],
|
|
email=user_data["email"],
|
|
password=user_data["password"],
|
|
company=jumbo_company,
|
|
is_company_admin=user_data["is_company_admin"],
|
|
)
|
|
users_created += 1
|
|
|
|
if users_created:
|
|
self.stdout.write(self.style.SUCCESS(f"✓ Created {users_created} Jumbo users"))
|
|
else:
|
|
self.stdout.write(" Jumbo users already exist")
|
|
|
|
# 4. Link External Data Source to Jumbo company
|
|
try:
|
|
jumbo_ext_source = ExternalDataSource.objects.get(name="Jumbo API")
|
|
if not jumbo_ext_source.company:
|
|
jumbo_ext_source.company = jumbo_company
|
|
jumbo_ext_source.save()
|
|
self.stdout.write(self.style.SUCCESS("✓ Linked Jumbo API data source to company"))
|
|
else:
|
|
self.stdout.write(" Jumbo API data source already linked")
|
|
except ExternalDataSource.DoesNotExist:
|
|
self.stdout.write(
|
|
self.style.WARNING("⚠ Jumbo API external data source not found. Create it in admin first.")
|
|
)
|
|
|
|
# 5. Link existing chat sessions to Jumbo company
|
|
unlinked_sessions = ChatSession.objects.filter(company__isnull=True)
|
|
if unlinked_sessions.exists():
|
|
count = unlinked_sessions.update(company=jumbo_company)
|
|
self.stdout.write(self.style.SUCCESS(f"✓ Linked {count} existing chat sessions to Jumbo company"))
|
|
else:
|
|
self.stdout.write(" All chat sessions already linked to companies")
|
|
|
|
# 6. Summary
|
|
total_sessions = ChatSession.objects.filter(company=jumbo_company).count()
|
|
total_users = CustomUser.objects.filter(company=jumbo_company).count()
|
|
|
|
self.stdout.write(
|
|
self.style.SUCCESS(
|
|
f"\n✓ Setup complete!"
|
|
f"\n Company: {jumbo_company.name}"
|
|
f"\n Users: {total_users} (including {1 if admin_created or CustomUser.objects.filter(username='jumbo_admin').exists() else 0} admin)"
|
|
f"\n Chat sessions: {total_sessions}"
|
|
)
|
|
)
|
|
self.stdout.write("\nLogin as jumbo_admin/jumbo123 to view the dashboard with Jumbo data.")
|