Implemented relationship between Pagarme Item and Subscription Type

renzon · renzon · commit 989c47a2a694 · 2021-06-05T13:17:08.000-03:00
close #3761
diff --git a/pythonpro/memberkit/admin.py b/pythonpro/memberkit/admin.py
@@ -1,17 +1,26 @@
 from django.contrib import admin
 from django.shortcuts import redirect
 from django.urls import path
+from django_pagarme.admin import PagarmeItemConfigAdmin
+from django_pagarme.models import PagarmeItemConfig
 
 # Register your models here.
 from pythonpro.memberkit import facade
-from pythonpro.memberkit.models import SubscriptionType
+from pythonpro.memberkit.models import SubscriptionType, PaymentItemConfigToSubscriptionType
+
+
+class PaymentItemConfigInline(admin.TabularInline):
+    extra = 1
+    model = PaymentItemConfigToSubscriptionType
 
 
 @admin.register(SubscriptionType)
 class SubscriptionTypeAdmin(admin.ModelAdmin):
     change_list_template = "memberkit/subscriptiontype/synchronize_button.html"
     fields = ['id', 'name']
     list_display = fields
+    readonly_fields = fields
+    inlines = [PaymentItemConfigInline]
 
     def get_urls(self):
         urls = super().get_urls()
@@ -29,5 +38,10 @@ def has_add_permission(self, request):
     def has_delete_permission(self, request, obj=None):
         return False
 
-    def has_change_permission(self, request, obj=None):
-        return False
+
+admin.site.unregister(PagarmeItemConfig)
+
+
+@admin.register(PagarmeItemConfig)
+class NewPagarmeItemConfigAdmin(PagarmeItemConfigAdmin):
+    inlines = [PaymentItemConfigInline]
diff --git a/pythonpro/memberkit/migrations/0002_create_relationship_with_payment_config_item.py b/pythonpro/memberkit/migrations/0002_create_relationship_with_payment_config_item.py
@@ -0,0 +1,30 @@
+# Generated by Django 3.2.4 on 2021-06-05 15:46
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ('django_pagarme', '0004_pagarme_item_config_available_until'),
+        ('memberkit', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AlterModelOptions(
+            name='subscriptiontype',
+            options={'verbose_name': 'Tipo de Assinatura', 'verbose_name_plural': 'Tipos de Assinaturas'},
+        ),
+        migrations.CreateModel(
+            name='PaymentItemConfigToSubscriptionType',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('payment_item', models.OneToOneField(on_delete=django.db.models.deletion.CASCADE,
+                                                      related_name='subscription_type_relation',
+                                                      to='django_pagarme.pagarmeitemconfig')),
+                ('subscription_type',
+                 models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='payment_items_relation',
+                                   to='memberkit.subscriptiontype')),
+            ],
+        ),
+    ]
diff --git a/pythonpro/memberkit/models.py b/pythonpro/memberkit/models.py
@@ -11,3 +11,19 @@ def __str__(self):
     class Meta:
         verbose_name = 'Tipo de Assinatura'
         verbose_name_plural = 'Tipos de Assinaturas'
+
+
+class PaymentItemConfigToSubscriptionType(models.Model):
+    payment_item = models.OneToOneField('django_pagarme.PagarmeItemConfig', null=False, on_delete=models.CASCADE,
+                                        related_name='subscription_type_relation')
+    subscription_type = models.ForeignKey(SubscriptionType, null=False, on_delete=models.CASCADE,
+                                          related_name='payment_items_relation')
+
+    def __str__(self):
+        return f'{self.payment_item} -> {self.subscription_type}'
+
+# class Subscription(models.Model):
+#     created_at = models.DateTimeField(auto_now_add=True)
+#     updated_at = models.DateTimeField(auto_now_add=True)
+#     payment = models.OneToOneField('PagarmePayment', on_delete=models.DO_NOTHING, )
+#     subscription_type = models.OneToOneField(SubscriptionType, null=False, on_delete=models.DO_NOTHING)
