Add changes to billing plans

app_users/migrations/0026_alter_appusertransaction_plan.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1.3 on 2025-03-25 11:06
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('app_users', '0025_remove_appuser_handle'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='appusertransaction',
+            name='plan',
+            field=models.IntegerField(blank=True, choices=[(1, 'Basic Plan'), (2, 'Premium Plan'), (4, 'Creator'), (5, 'Business (2024)'), (3, 'Pay as you go'), (7, 'Business'), (6, 'Enterprise')], default=None, help_text="User's plan at the time of this transaction.", null=True),
+        ),
+    ]

app_users/models.py

@@ -17,6 +17,7 @@
 
 if typing.TYPE_CHECKING:
     from workspaces.models import Workspace
+    from phonenumber_field.phonenumber import PhoneNumber
 
 
 class AppUserQuerySet(models.QuerySet):
@@ -164,7 +165,7 @@ def full_name(
         elif self.email:
             name = self.email.split("@")[0]
         elif self.phone_number:
-            name = str(self.phone_number)
+            name = obscure_phone_number(self.phone_number)
         else:
             return fallback
         if current_user and self == current_user:
@@ -253,6 +254,9 @@ def get_anonymous_token(self):
     def get_photo(self) -> str:
         return self.photo_url or get_placeholder_profile_image(self.uid)
 
+    def is_admin(self) -> bool:
+        return bool(self.email and self.email in settings.ADMIN_EMAILS)
+
 
 class TransactionReason(models.IntegerChoices):
     DEDUCT = 1, "Deduct"
@@ -391,3 +395,20 @@ def payment_provider_url(self) -> str | None:
 def get_placeholder_profile_image(seed: str) -> str:
     hash = hashlib.md5(seed.encode()).hexdigest()
     return f"https://gravatar.com/avatar/{hash}?d=robohash&size=150"
+
+
+def obscure_phone_number(phone_number: "PhoneNumber") -> str:
+    """
+    Obscure the phone number by replacing the middle digits with asterisks.
+    """
+    country_code = phone_number.country_code
+    national_number = str(phone_number.national_number)
+
+    return "".join(
+        [
+            (country_code and f"+{country_code}" or ""),
+            national_number[:3],
+            "*" * len(national_number[3:-3]),
+            national_number[-3:],
+        ]
+    )

bots/admin.py

@@ -322,15 +322,15 @@ class PublishedRunVersionInline(admin.TabularInline):
 class PublishedRunAdmin(admin.ModelAdmin):
     list_display = [
         "__str__",
-        "visibility",
+        "public_access",
         "view_user",
         "open_in_gooey",
         "linked_saved_run",
         "view_runs",
         "created_at",
         "updated_at",
     ]
-    list_filter = ["workflow", "visibility", "created_by__is_paying"]
+    list_filter = ["workflow", "public_access", "created_by__is_paying"]
     search_fields = ["workflow", "published_run_id", "title", "notes"]
     autocomplete_fields = ["saved_run", "created_by", "last_edited_by", "workspace"]
     readonly_fields = [

bots/migrations/0049_publishedrun_publishedrunversion.py

@@ -3,7 +3,7 @@
 from django.db import migrations, models
 import django.db.models.deletion
 
-from bots.models import PublishedRunVisibility
+from bots.models import WorkflowAccessLevel
 from daras_ai_v2.crypto import get_random_doc_id
 
 
@@ -30,7 +30,7 @@ def create_published_run_from_example(
         saved_run=saved_run,
         title=saved_run.page_title,
         notes=saved_run.page_notes,
-        visibility=PublishedRunVisibility.PUBLIC,
+        visibility=WorkflowAccessLevel.FIND_AND_VIEW,
         is_approved_example=not saved_run.hidden,
     )
     set_field_attribute(published_run, "created_at", auto_now_add=False)
@@ -48,7 +48,7 @@ def create_published_run_from_example(
         changed_by=user,
         title=saved_run.page_title,
         notes=saved_run.page_notes,
-        visibility=PublishedRunVisibility.PUBLIC,
+        visibility=WorkflowAccessLevel.FIND_AND_VIEW,
     )
     set_field_attribute(published_run, "created_at", auto_now_add=False)
     version.created_at = saved_run.updated_at

bots/migrations/0095_remove_publishedrun_bots_publis_updated_af6358_idx_and_more.py

@@ -0,0 +1,51 @@
+# Generated by Django 5.1.3 on 2025-04-04 08:30
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('app_users', '0026_alter_appusertransaction_plan'),
+        ('bots', '0094_publishedrun_photo_url_workflowmetadata_emoji'),
+        ('workspaces', '0010_workspace_banner_url'),
+    ]
+
+    operations = [
+        migrations.RemoveIndex(
+            model_name='publishedrun',
+            name='bots_publis_updated_af6358_idx',
+        ),
+        migrations.RemoveIndex(
+            model_name='publishedrun',
+            name='bots_publis_is_appr_a989ec_idx',
+        ),
+        migrations.AddField(
+            model_name='publishedrun',
+            name='workspace_access',
+            field=models.IntegerField(choices=[(1, 'View Only'), (2, 'Find And View'), (4, 'Edit'), (3, 'Internal')], default=4),
+        ),
+        migrations.AddField(
+            model_name='publishedrunversion',
+            name='workspace_access',
+            field=models.IntegerField(choices=[(1, 'View Only'), (2, 'Find And View'), (4, 'Edit'), (3, 'Internal')], default=4),
+        ),
+        migrations.AlterField(
+            model_name='publishedrun',
+            name='visibility',
+            field=models.IntegerField(choices=[(1, 'View Only'), (2, 'Find And View'), (4, 'Edit'), (3, 'Internal')], default=2),
+        ),
+        migrations.AlterField(
+            model_name='publishedrunversion',
+            name='visibility',
+            field=models.IntegerField(choices=[(1, 'View Only'), (2, 'Find And View'), (4, 'Edit'), (3, 'Internal')], default=1),
+        ),
+        migrations.AddIndex(
+            model_name='publishedrun',
+            index=models.Index(fields=['is_approved_example', 'visibility', 'workspace_access', 'published_run_id', 'updated_at', 'workflow', 'example_priority'], name='bots_publis_is_appr_8c8651_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='publishedrun',
+            index=models.Index(fields=['-updated_at', 'workspace', 'created_by', 'visibility', 'workspace_access'], name='bots_publis_updated_31d8a4_idx'),
+        ),
+    ]

bots/migrations/0096_remove_publishedrun_bots_publis_is_appr_8c8651_idx_and_more.py

@@ -0,0 +1,41 @@
+# Generated by Django 5.1.3 on 2025-04-04 08:31
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('app_users', '0026_alter_appusertransaction_plan'),
+        ('bots', '0095_remove_publishedrun_bots_publis_updated_af6358_idx_and_more'),
+        ('workspaces', '0010_workspace_banner_url'),
+    ]
+
+    operations = [
+        migrations.RemoveIndex(
+            model_name='publishedrun',
+            name='bots_publis_is_appr_8c8651_idx',
+        ),
+        migrations.RemoveIndex(
+            model_name='publishedrun',
+            name='bots_publis_updated_31d8a4_idx',
+        ),
+        migrations.RenameField(
+            model_name='publishedrun',
+            old_name='visibility',
+            new_name='public_access',
+        ),
+        migrations.RenameField(
+            model_name='publishedrunversion',
+            old_name='visibility',
+            new_name='public_access',
+        ),
+        migrations.AddIndex(
+            model_name='publishedrun',
+            index=models.Index(fields=['is_approved_example', 'public_access', 'workspace_access', 'published_run_id', 'updated_at', 'workflow', 'example_priority'], name='bots_publis_is_appr_644b75_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='publishedrun',
+            index=models.Index(fields=['-updated_at', 'workspace', 'created_by', 'public_access', 'workspace_access'], name='bots_publis_updated_fcf021_idx'),
+        ),
+    ]