feat: add database models for supplier profiles, onboarding drafts, and categories
All checks were successful
Deploy Beta (NATIVE) / deploy (push) Successful in 52s

This commit is contained in:
vickytechkey 2026-11-10 19:25:25 +05:30
parent 1d707e674b
commit 580dab5b42
4 changed files with 245 additions and 0 deletions

View file

@ -0,0 +1,48 @@
# Generated by Django 6.1 on 2026-11-10 13:44
from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
('api', '0021_product_coordinates_product_geographical_origin_and_more'),
]
operations = [
migrations.AddField(
model_name='supplieronboardingdraft',
name='gi_seller_type',
field=models.CharField(default='non_gi', max_length=50),
),
migrations.AddField(
model_name='supplieronboardingdraft',
name='is_exclusive_gi',
field=models.BooleanField(default=False),
),
migrations.AddField(
model_name='supplierprofile',
name='gi_certificate_number',
field=models.CharField(blank=True, max_length=100, null=True),
),
migrations.AddField(
model_name='supplierprofile',
name='gi_registry_verified',
field=models.BooleanField(default=False),
),
migrations.AddField(
model_name='supplierprofile',
name='gi_seller_type',
field=models.CharField(choices=[('exclusive', 'Exclusive GI Only'), ('mixed', 'Mixed Catalog'), ('non_gi', 'Non-GI Craft')], default='non_gi', max_length=50),
),
migrations.AddField(
model_name='supplierprofile',
name='guild_status',
field=models.CharField(blank=True, max_length=255, null=True),
),
migrations.AddField(
model_name='supplierprofile',
name='is_exclusive_gi',
field=models.BooleanField(default=False),
),
]

View file

@ -119,6 +119,13 @@ class SupplierProfile(models.Model):
featured_tagline = models.CharField(max_length=255, blank=True, null=True)
social_links = models.JSONField(default=dict, blank=True, null=True)
# GI Accreditation & Storefront Status
gi_seller_type = models.CharField(max_length=50, default='non_gi', choices=[('exclusive', 'Exclusive GI Only'), ('mixed', 'Mixed Catalog'), ('non_gi', 'Non-GI Craft')])
is_exclusive_gi = models.BooleanField(default=False)
gi_certificate_number = models.CharField(max_length=100, blank=True, null=True)
guild_status = models.CharField(max_length=255, blank=True, null=True)
gi_registry_verified = models.BooleanField(default=False)
def __block__(self):
return
@ -201,6 +208,8 @@ class SupplierOnboardingDraft(models.Model):
artisan_avatar = models.TextField(blank=True, null=True)
craft_lineage_years = models.IntegerField(default=0)
social_links = models.JSONField(default=dict, blank=True, null=True)
gi_seller_type = models.CharField(max_length=50, default='non_gi')
is_exclusive_gi = models.BooleanField(default=False)
# Contact
business_email = models.EmailField(blank=True, null=True)

View file

@ -673,6 +673,10 @@ class ProfileView(APIView):
draft.workshop_location = request.data.get('workshop_location', draft.workshop_location)
draft.artisan_avatar = request.data.get('artisan_avatar', draft.artisan_avatar)
draft.featured_tagline = request.data.get('featured_tagline', draft.featured_tagline)
if 'gi_seller_type' in request.data:
draft.gi_seller_type = request.data.get('gi_seller_type')
if 'is_exclusive_gi' in request.data:
draft.is_exclusive_gi = request.data.get('is_exclusive_gi')
if 'sample_product_data' in request.data:
draft.sample_product_data = request.data.get('sample_product_data')
if 'social_links' in request.data:
@ -764,6 +768,13 @@ class ProfileView(APIView):
if social_links:
profile.social_links = social_links
# GI Accreditation & Storefront Status
profile.gi_seller_type = pick_val('gi_seller_type', getattr(profile, 'gi_seller_type', 'non_gi'))
profile.is_exclusive_gi = pick_val('is_exclusive_gi', getattr(profile, 'is_exclusive_gi', False))
profile.gi_certificate_number = pick_val('gi_certificate_number', getattr(profile, 'gi_certificate_number', None))
profile.guild_status = pick_val('guild_status', getattr(profile, 'guild_status', None))
profile.gi_registry_verified = pick_val('gi_registry_verified', getattr(profile, 'gi_registry_verified', False))
# Store auto-approval upon onboarding completion
profile.is_profile_complete = True
profile.status = 'approved'

177
test_gi_workflow.py Normal file
View file

@ -0,0 +1,177 @@
"""
Test suite to thoroughly verify the GI accreditation logic, default state,
and CRM approval/rejection sync workflow.
"""
import os
import sys
# Configure minimal in-memory Django environment
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'config.settings')
import django
from django.conf import settings
# Override databases to in-memory SQLite for instantaneous local test execution
settings.DATABASES = {
'default': {
'ENGINE': 'django.db.backends.sqlite3',
'NAME': ':memory:',
},
'product_db': {
'ENGINE': 'django.db.backends.sqlite3',
'NAME': ':memory:',
},
'seller_db': {
'ENGINE': 'django.db.backends.sqlite3',
'NAME': ':memory:',
}
}
django.setup()
from django.core.management import call_command
from django.contrib.auth.models import User
from api.models import SupplierProfile, SupplierOnboardingDraft
from api.serializers import SupplierProfileSerializer
print("Setting up in-memory database...")
call_command('migrate', verbosity=0)
# ==============================================================================
# TEST 1: Default seller values are Non-GI Craft ('non_gi')
# ==============================================================================
print("\n--- TEST 1: Checking default GI accreditation fields ---")
user1 = User.objects.create(username="artisan_ramesh", email="ramesh@example.com")
profile1 = SupplierProfile.objects.create(
user=user1,
store_name="Ramesh Weaves",
phone="9876543210"
)
assert profile1.gi_seller_type == 'non_gi', f"Expected default 'non_gi', got {profile1.gi_seller_type}"
assert profile1.is_exclusive_gi is False, f"Expected is_exclusive_gi=False, got {profile1.is_exclusive_gi}"
assert profile1.gi_registry_verified is False, f"Expected gi_registry_verified=False, got {profile1.gi_registry_verified}"
assert profile1.gi_certificate_number is None
assert profile1.guild_status is None
print("✓ TEST 1 PASSED: New seller profile defaults to 'non_gi' and unverified.")
# ==============================================================================
# TEST 2: SupplierOnboardingDraft defaults
# ==============================================================================
print("\n--- TEST 2: Checking SupplierOnboardingDraft defaults ---")
draft = SupplierOnboardingDraft.objects.create(
user=user1,
phone="9876543210"
)
assert draft.gi_seller_type == 'non_gi', f"Expected default 'non_gi', got {draft.gi_seller_type}"
assert draft.is_exclusive_gi is False, f"Expected is_exclusive_gi=False, got {draft.is_exclusive_gi}"
print("✓ TEST 2 PASSED: SupplierOnboardingDraft defaults to 'non_gi' and False.")
# ==============================================================================
# TEST 3: Serializer serialization of all 5 GI fields
# ==============================================================================
print("\n--- TEST 3: Checking SupplierProfileSerializer output ---")
profile1.gi_seller_type = 'exclusive'
profile1.is_exclusive_gi = True
profile1.gi_certificate_number = 'GI-TN-2005-0012'
profile1.guild_status = 'Kanchipuram Silk Weavers Guild'
profile1.gi_registry_verified = True
profile1.save()
serialized = SupplierProfileSerializer(profile1).data
assert serialized['gi_seller_type'] == 'exclusive', f"Serialized error: {serialized.get('gi_seller_type')}"
assert serialized['is_exclusive_gi'] is True, f"Serialized error: {serialized.get('is_exclusive_gi')}"
assert serialized['gi_certificate_number'] == 'GI-TN-2005-0012', f"Serialized error: {serialized.get('gi_certificate_number')}"
assert serialized['guild_status'] == 'Kanchipuram Silk Weavers Guild', f"Serialized error: {serialized.get('guild_status')}"
assert serialized['gi_registry_verified'] is True, f"Serialized error: {serialized.get('gi_registry_verified')}"
print("✓ TEST 3 PASSED: Serializer correctly exposes all 5 GI accreditation fields.")
# ==============================================================================
# TEST 4: Simulating CRM Product Approval & Seller GI Auto-Upgrade
# ==============================================================================
print("\n--- TEST 4: Verifying CRM Sync Logic ---")
class MockProduct:
def __init__(self, id, supplier_id, title, is_gi_tagged, gi_tag_name, geographical_origin, status="approved"):
self.id = id
self.supplier_id = supplier_id
self.title = title
self.is_gi_tagged = is_gi_tagged
self.gi_tag_name = gi_tag_name
self.geographical_origin = geographical_origin
self.status = status
self.is_active = (status == "approved")
def simulate_crm_sync(seller, products):
approved_prods = [p for p in products if p.status == "approved" and p.is_active]
total_approved = len(approved_prods)
gi_approved = [p for p in approved_prods if p.is_gi_tagged]
gi_count = len(gi_approved)
if gi_count > 0:
first_gi = gi_approved[0]
gi_name = first_gi.gi_tag_name or "Govt. of India GI Registry Verified"
if gi_count == total_approved and total_approved > 0:
seller.gi_seller_type = "exclusive"
seller.is_exclusive_gi = True
else:
seller.gi_seller_type = "mixed"
seller.is_exclusive_gi = False
seller.gi_registry_verified = True
seller.gi_certificate_number = gi_name
seller.guild_status = first_gi.geographical_origin or "Certified GI Artisan Maker"
else:
seller.gi_seller_type = "non_gi"
seller.is_exclusive_gi = False
seller.gi_registry_verified = False
seller.save()
# Scenario 4A: Seller has only standard craft product approved
prods_non_gi = [
MockProduct(1, profile1.id, "Terracotta Pot", is_gi_tagged=False, gi_tag_name=None, geographical_origin=None, status="approved")
]
simulate_crm_sync(profile1, prods_non_gi)
profile1.refresh_from_db()
assert profile1.gi_seller_type == "non_gi"
assert profile1.is_exclusive_gi is False
assert profile1.gi_registry_verified is False
print("✓ TEST 4A PASSED: Standard non-GI product keeps seller as 'non_gi'.")
# Scenario 4B: Seller has a GI product approved -> Upgrades to 'exclusive'
prods_gi_only = [
MockProduct(2, profile1.id, "Kanchipuram Silk Saree", is_gi_tagged=True, gi_tag_name="GI Tag #26 (1999)", geographical_origin="Kanchipuram, Tamil Nadu", status="approved")
]
simulate_crm_sync(profile1, prods_gi_only)
profile1.refresh_from_db()
assert profile1.gi_seller_type == "exclusive"
assert profile1.is_exclusive_gi is True
assert profile1.gi_registry_verified is True
assert profile1.gi_certificate_number == "GI Tag #26 (1999)"
print("✓ TEST 4B PASSED: 100% GI approved products upgrades seller to 'exclusive' with verified GI badge.")
# Scenario 4C: Mixed catalog (1 GI product + 1 non-GI product)
prods_mixed = [
MockProduct(2, profile1.id, "Kanchipuram Silk Saree", is_gi_tagged=True, gi_tag_name="GI Tag #26 (1999)", geographical_origin="Kanchipuram, Tamil Nadu", status="approved"),
MockProduct(1, profile1.id, "Terracotta Pot", is_gi_tagged=False, gi_tag_name=None, geographical_origin=None, status="approved")
]
simulate_crm_sync(profile1, prods_mixed)
profile1.refresh_from_db()
assert profile1.gi_seller_type == "mixed"
assert profile1.is_exclusive_gi is False
assert profile1.gi_registry_verified is True
print("✓ TEST 4C PASSED: Combination of GI and non-GI products categorizes seller as 'mixed'.")
# Scenario 4D: Admin rejects the GI product -> Reverts to 'non_gi'
prods_rejected = [
MockProduct(2, profile1.id, "Kanchipuram Silk Saree", is_gi_tagged=True, gi_tag_name="GI Tag #26 (1999)", geographical_origin="Kanchipuram, Tamil Nadu", status="rejected"),
MockProduct(1, profile1.id, "Terracotta Pot", is_gi_tagged=False, gi_tag_name=None, geographical_origin=None, status="approved")
]
simulate_crm_sync(profile1, prods_rejected)
profile1.refresh_from_db()
assert profile1.gi_seller_type == "non_gi"
assert profile1.is_exclusive_gi is False
assert profile1.gi_registry_verified is False
print("✓ TEST 4D PASSED: Rejection of GI products reverts seller back to 'non_gi'.")
print("\n=======================================================")
print(" ALL 4 TEST SCENARIOS PASSED WITH 100% SUCCESS! ")
print("=======================================================\n")