connecting to postgresql

Author: TechWithTy

backend/apps/authentication/migrations/0001_initial.py

@@ -0,0 +1,60 @@
+# Generated by Django 4.2.10 on 2025-03-27 01:32
+
+from django.conf import settings
+import django.contrib.auth.models
+import django.contrib.auth.validators
+from django.db import migrations, models
+import django.db.models.deletion
+import django.utils.timezone
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        ('auth', '0012_alter_user_first_name_max_length'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='CustomUser',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('password', models.CharField(max_length=128, verbose_name='password')),
+                ('last_login', models.DateTimeField(blank=True, null=True, verbose_name='last login')),
+                ('is_superuser', models.BooleanField(default=False, help_text='Designates that this user has all permissions without explicitly assigning them.', verbose_name='superuser status')),
+                ('username', models.CharField(error_messages={'unique': 'A user with that username already exists.'}, help_text='Required. 150 characters or fewer. Letters, digits and @/./+/-/_ only.', max_length=150, unique=True, validators=[django.contrib.auth.validators.UnicodeUsernameValidator()], verbose_name='username')),
+                ('first_name', models.CharField(blank=True, max_length=150, verbose_name='first name')),
+                ('last_name', models.CharField(blank=True, max_length=150, verbose_name='last name')),
+                ('email', models.EmailField(blank=True, max_length=254, verbose_name='email address')),
+                ('is_staff', models.BooleanField(default=False, help_text='Designates whether the user can log into this admin site.', verbose_name='staff status')),
+                ('is_active', models.BooleanField(default=True, help_text='Designates whether this user should be treated as active. Unselect this instead of deleting accounts.', verbose_name='active')),
+                ('date_joined', models.DateTimeField(default=django.utils.timezone.now, verbose_name='date joined')),
+                ('supabase_uid', models.CharField(blank=True, max_length=255, null=True, unique=True)),
+                ('oauth_provider', models.CharField(blank=True, max_length=50, null=True)),
+                ('metadata', models.JSONField(blank=True, default=dict, null=True)),
+                ('groups', models.ManyToManyField(blank=True, help_text='The groups this user belongs to. A user will get all permissions granted to each of their groups.', related_name='user_set', related_query_name='user', to='auth.group', verbose_name='groups')),
+                ('user_permissions', models.ManyToManyField(blank=True, help_text='Specific permissions for this user.', related_name='user_set', related_query_name='user', to='auth.permission', verbose_name='user permissions')),
+            ],
+            options={
+                'verbose_name': 'User',
+                'verbose_name_plural': 'Users',
+                'swappable': 'AUTH_USER_MODEL',
+            },
+            managers=[
+                ('objects', django.contrib.auth.models.UserManager()),
+            ],
+        ),
+        migrations.CreateModel(
+            name='UserData',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('profile_data', models.JSONField(blank=True, default=dict, null=True)),
+                ('preferences', models.JSONField(blank=True, default=dict, null=True)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('user', models.OneToOneField(on_delete=django.db.models.deletion.CASCADE, related_name='user_data', to=settings.AUTH_USER_MODEL)),
+            ],
+        ),
+    ]

backend/apps/authentication/migrations/__init__.py

@@ -1,5 +1,5 @@
 from django.db import models, transaction
-from django.contrib.auth.models import User
+from django.conf import settings
 from django.utils.translation import gettext_lazy as _
 import logging
 import uuid
@@ -27,7 +27,7 @@ class CreditTransaction(models.Model):
         editable=False
     )
     user = models.ForeignKey(
-        User,
+        settings.AUTH_USER_MODEL,
         on_delete=models.CASCADE,
         related_name='credit_transactions'
     )
@@ -113,7 +113,7 @@ class CreditHold(models.Model):
         editable=False
     )
     user = models.ForeignKey(
-        User,
+        settings.AUTH_USER_MODEL,
         on_delete=models.CASCADE,
         related_name='credit_holds'
     )

backend/apps/credits/models.py

@@ -105,7 +105,7 @@ def _get_required_credits(self, request: Request) -> int:
             pattern = rate.endpoint_path
             # Convert the endpoint pattern to a regex pattern
             # Replace {id} or similar placeholders with regex groups
-            regex_pattern = pattern.replace('/', '\/')
+            regex_pattern = pattern.replace('/', r'\/')
             regex_pattern = re.sub(r'\{[^}]+\}', r'[^\/]+', regex_pattern)
 
             if re.match(regex_pattern, path):

backend/apps/credits/throttling.py

@@ -1,15 +1,15 @@
 from django.urls import path, include
 from rest_framework.routers import DefaultRouter
 
-from . import base
+from . import views
 
 router = DefaultRouter()
 router.register(
-    r"transactions", base.CreditTransactionViewSet, basename="credit-transaction"
+    r"transactions", views.CreditTransactionViewSet, basename="credit-transaction"
 )
-router.register(r"rates", base.CreditUsageRateViewSet, basename="credit-rate")
+router.register(r"rates", views.CreditUsageRateViewSet, basename="credit-rate")
 
 urlpatterns = [
-    path("credits/", base.get_credit_balance, name="credit-balance"),
+    path("credits/", views.get_credit_balance, name="credit-balance"),
     path("credits/", include(router.urls)),
 ]

backend/apps/credits/urls.py

@@ -4,7 +4,7 @@
 import json
 
 from django.conf import settings
-from django.contrib.auth.models import User
+from django.contrib.auth import get_user_model
 from rest_framework import viewsets, status, permissions
 from rest_framework.decorators import action, api_view, permission_classes
 from rest_framework.request import Request
@@ -20,6 +20,9 @@
 # Initialize the Supabase Auth Service
 auth_service = SupabaseAuthService()
 
+# Get the custom user model
+User = get_user_model()
+
 
 class IsAdminOrSelf(permissions.BasePermission):
     """

backend/apps/users/base.py

@@ -1,5 +1,5 @@
 from django.db import models, transaction
-from django.contrib.auth.models import User
+from django.conf import settings
 from django.utils.translation import gettext_lazy as _
 
 class UserProfile(models.Model):
@@ -10,7 +10,7 @@ class UserProfile(models.Model):
     that are specific to our application, including Supabase-related information.
     """
     user = models.OneToOneField(
-        User, 
+        settings.AUTH_USER_MODEL, 
         on_delete=models.CASCADE, 
         related_name='profile'
     )

backend/apps/users/models.py

@@ -0,0 +1,37 @@
+from django.conf import settings
+from django.db import models
+
+
+
+# The Profile model extends the default user model with additional fields for AI-enabled features
+# and credit management. This ensures that each user can have unique AI-related settings and credit balance.
+
+class Profile(models.Model):
+    user = models.OneToOneField(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name='profile')
+    first_name = models.CharField(max_length=50, blank=True, null=True, help_text="User's first name")
+    last_name = models.CharField(max_length=50, blank=True, null=True, help_text="User's last name")
+    email = models.EmailField(blank=True, null=True, help_text="User's email address")
+    phone_number = models.CharField(max_length=20, blank=True, null=True, help_text="User's contact phone number")
+    source = models.CharField(max_length=100, blank=True, null=True, help_text="Signup source or context")
+    contextual_info = models.TextField(blank=True, null=True, help_text="Additional contextual information for business prompting")
+    ai_enabled = models.BooleanField(default=False, help_text='Enable AI features for the user')
+    credits = models.PositiveIntegerField(default=0, help_text='Credit balance for AI execution')
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+
+    def __str__(self):
+        return f"Profile for {self.user.username}"
+
+
+
+# The CreditTransaction model records credit transactions related to AI usage. 
+# This allows for tracking and auditing credit usages or additions over time.
+
+class CreditTransaction(models.Model):
+    profile = models.ForeignKey(Profile, on_delete=models.CASCADE, related_name='transactions')
+    amount = models.IntegerField(help_text='Amount of credits deducted (negative) or added (positive)')
+    description = models.TextField(blank=True, null=True, help_text='Reason for this transaction')
+    created_at = models.DateTimeField(auto_now_add=True)
+
+    def __str__(self):
+        return f"Transaction of {self.amount} for {self.profile.user.username} at {self.created_at}"

backend/apps/users/models/profile.py

@@ -0,0 +1,19 @@
+from django.conf import settings
+from django.db import models
+from .profile import Profile
+
+
+# The UserContext model captures contextual business flow information for agentive operations.
+# This model supports dynamic business operations by storing relevant context data in a JSON field,
+# which can be used to tailor AI-driven business processes.
+
+
+class UserContext(models.Model):
+    profile = models.ForeignKey(Profile, on_delete=models.CASCADE, related_name='user_contexts')
+    business_flow = models.CharField(max_length=100, help_text='Name or type of the business flow')
+    context_data = models.JSONField(blank=True, null=True, help_text='Detailed context information for agentive business flows')
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+
+    def __str__(self):
+        return f"Context for {self.profile.user.username} - {self.business_flow}"

backend/apps/users/models/user_context.py

@@ -3,8 +3,11 @@
 from rest_framework.routers import DefaultRouter
 
 from . import base
-from .base import auth_view, client_view, database_view, edge_functions_view, health_check, realtime_view, storage_view, script_view
+from .views import auth_view, client_view, database_view, edge_functions_view, realtime_view, storage_view
+from .views.creditable_views import utility_view
 
+# Import health check from base if it exists there, otherwise we'll need to find it
+from ..authentication.views import health_check
 router = DefaultRouter()
 router.register(r'users', base.UserViewSet)