chore: site init added

Author: Bishwas-py

django_backend/django_backend/settings.py

@@ -131,5 +131,3 @@
 # https://docs.djangoproject.com/en/4.2/ref/settings/#default-auto-field
 
 DEFAULT_AUTO_FIELD = 'django.db.models.BigAutoField'
-
-SITE_ID = 1

django_backend/home/apps.py

@@ -1,16 +1,31 @@
 import sys
-
 from django.apps import AppConfig
+from django.core.signals import request_started
 
 
 class HomeConfig(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'home'
 
    def ready(self):
+      # Skip database initialization for management commands
+      if self.should_skip_initialization():
+         return
+
+      # Connect to the request_started signal instead of running directly
+      request_started.connect(self.initialize_site_data, dispatch_uid="initialize_site_data")
+
+   def should_skip_initialization(self):
+      management_commands = {
+         'makemigrations', 'migrate', 'collectstatic',
+         'createsuperuser', 'dbshell', 'test'
+      }
+      return any(cmd in sys.argv for cmd in management_commands)
+
+   def initialize_site_data(self, sender, **kwargs):
+      # Disconnect the signal after first run to prevent multiple executions
+      request_started.disconnect(self.initialize_site_data, dispatch_uid="initialize_site_data")
+
+      # Import and run the initialization
       from home.startup import site_data_db_startup
-      if ('makemigrations' not in sys.argv and 'migrate' not in sys.argv
-        and 'collectstatic' not in sys.argv
-        and 'createsuperuser' not in sys.argv
-        and 'dbshell' not in sys.argv):
-         site_data_db_startup()
+      site_data_db_startup()

django_backend/home/startup.py

@@ -1,54 +1,116 @@
-import logging
+from typing import Dict, Any
+from django.db import transaction
+from django.contrib.sites.models import Site
+from dataclasses import dataclass
+from utils.c_logging import logger
 
-SITE_DEFAULT_DATA = {
-   'title': 'Djapy - Django and Python Supremacy',
-   'subtitle': 'We love django and python, so should you',
-   'description': 'freelancing, coding, indie dev, app dev',
-   'domain': 'djapy.io',
-   'name': 'Djapy',
-   'keywords': 'django, python, djapy, coding, indie dev, app dev',
-   'ads_on': False,
-}
+try:
+   from config.env import DOMAIN
+except ImportError:
+   DOMAIN = 'localhost:5173'
 
 
-def site_data_db_startup():
-   """
-   This function is called to store default site data in the database.
-   """
-   from django.contrib.sites.models import Site
-   from home.models import SiteData
+@dataclass
+class SiteDefaults:
+   title: str = 'Djapy - Django and Python Supremacy'
+   subtitle: str = 'We love django and python, so should you'
+   description: str = 'freelancing, coding, indie dev, app dev'
+   domain: str = DOMAIN
+   name: str = 'Djapy'
+   keywords: str = 'django, python, djapy, coding, indie dev, app dev'
+   ads_on: bool = False
 
-   site = Site.objects.first()
-   if not site:
-      print("Creating site")
-      Site.objects.create(
-         domain=SITE_DEFAULT_DATA['domain'],
-         name=SITE_DEFAULT_DATA['name']
-      )
-   else:
-      print(f"Site already exists: {site.domain}")
+   def to_dict(self) -> Dict[str, Any]:
+      return {
+         field.name: getattr(self, field.name)
+         for field in self.__dataclass_fields__.values()
+      }
+
+
+def get_or_create_site(defaults: SiteDefaults) -> Site:
+   """Create or retrieve the default site."""
+   try:
+      site = Site.objects.first()
+      if not site:
+         logger.info("Creating new site")
+         site = Site.objects.create(
+            domain=defaults.domain,
+            name=defaults.name
+         )
+      else:
+         logger.info(f"Using existing site: {site.domain}")
+      return site
+   except Exception as e:
+      logger.error(f"Error creating/retrieving site: {str(e)}")
+      raise
 
-   if not SiteData.objects.filter(site=site).exists():
-      print("Creating site data")
-      SiteData.objects.create(
+
+def create_or_update_site_data(site: Site, defaults: SiteDefaults):
+   """Create or update site data."""
+   from home.models import SiteData
+
+   try:
+      site_data, created = SiteData.objects.get_or_create(
          site=site,
-         title=SITE_DEFAULT_DATA['title'],
-         subtitle=SITE_DEFAULT_DATA['subtitle'],
-         description=SITE_DEFAULT_DATA['description'],
-         keywords=SITE_DEFAULT_DATA['keywords'],
-         ads_on=SITE_DEFAULT_DATA['ads_on']
+         defaults={
+            'title': defaults.title,
+            'subtitle': defaults.subtitle,
+            'description': defaults.description,
+            'keywords': defaults.keywords,
+            'ads_on': defaults.ads_on
+         }
       )
-   else:
-      print("Site data already exists")
-
-   site_data = SiteData.objects.get(site=site)
-   print(
-      f"Site data:\nSite name: {site_data.site.name}"
-      f"\nSite title: {site_data.title}"
-      f"\nSite subtitle: {site_data.subtitle}"
-      f"\nSite description: {site_data.description}"
-      f"\nSite keywords: {site_data.keywords}"
-      f"\nSite ads on: {site_data.ads_on}"
-      f"\nSite domain: {site_data.site.domain}"
-      f"\nSite id: {site_data.site.id}"
-   )
+
+      if created:
+         logger.info("Created new site data")
+      else:
+         logger.info("Using existing site data")
+
+      return site_data
+   except Exception as e:
+      logger.error(f"Error creating/updating site data: {str(e)}")
+      raise
+
+
+def log_site_data(site_data) -> None:
+   """Log site data in a structured format."""
+   try:
+      info = {
+         'Site name': site_data.site.name,
+         'Site title': site_data.title,
+         'Site subtitle': site_data.subtitle,
+         'Site description': site_data.description,
+         'Site keywords': site_data.keywords,
+         'Site ads on': site_data.ads_on,
+         'Site domain': site_data.site.domain,
+         'Site id': site_data.site.id
+      }
+
+      logger.info("Site data configuration:")
+      for key, value in info.items():
+         logger.info(f"{key}: {value}")
+   except Exception as e:
+      logger.error(f"Error logging site data: {str(e)}")
+
+
+@transaction.atomic
+def site_data_db_startup() -> None:
+   """
+   Initialize or update site data in the database.
+   Uses atomic transaction to ensure data consistency.
+   """
+   try:
+      defaults = SiteDefaults()
+
+      # Get or create site
+      site = get_or_create_site(defaults)
+
+      # Create or update site data
+      site_data = create_or_update_site_data(site, defaults)
+
+      # Log the configuration
+      log_site_data(site_data)
+
+   except Exception as e:
+      logger.error(f"Failed to initialize site data: {str(e)}")
+      raise

django_backend/home/views.py

@@ -8,7 +8,7 @@
 @ensure_csrf_cookie
 @djapify
 def get_init_data(request) -> {200: InitDataSchema}:
-   site = Site.objects.get_current(request)
+   site = Site.objects.first()
    return {
       "site_data": site.sitedata,
       "current_user": request.user if request.user.is_authenticated else None