Merge pull request #6 from lk-geimfari/patch-1

Added one more disgusting view

Author: sobolevn

django/views.py

@@ -257,3 +257,101 @@ def create_payment(request):
 
             response = JsonResponse(post_params)
             return response
+
+        
+@staff_member_required
+def backup_to_csv(request):
+    data = {}
+    data['referral'] = ReferralPartner
+    data['user'] = UserProfile
+    data['exchange'] = CryptoExchange
+    data['payments'] = UserPayments
+    data['policy'] = InsurancePolicy
+    data['case'] = InsuranceCase
+    data['additional'] = AdditionalData
+    cursor = connection.cursor()
+    cursor.execute('''SELECT insurance_policy.id AS Policy_number,
+                        insurance_policy.request_date AS Policy_date,
+                        user_profile.first_name AS First_name,
+                        user_profile.last_name AS Last_name,
+                        user_profile.email AS Email,
+                        insurance_policy.start_date AS Start_date,
+                        insurance_policy.expiration_date AS Expiration_date,
+                        insurance_policy.expiration_date - \
+                        insurance_policy.start_date AS Number_of_days,
+                        crypto_exchange.name AS Crypto_exchange_name,
+                        crypto_exchange.coverage_limit AS Limit_BTC,
+                        insurance_policy.cover AS Insured_Limit,
+                        insurance_policy.fee AS Premium_paid,
+                        user_payments.amount AS User_paid,
+                        user_payments.currency AS User_currency,
+                        crypto_exchange.rate AS Premium_rate,
+                        user_payments.update_date AS Premium_payment_date,
+                        insurance_case.loss_value AS Outstanding_claim_BTC,
+                        insurance_case.incident_date AS Date_of_claim,
+                        insurance_case.refund_paid AS Paid_claim_BTC,
+                        insurance_case.request_date AS Date_of_claim_payment,
+                        insurance_policy.status AS Insurance_policy_status,
+                        user_payments.status AS User_payments_status,
+                        insurance_case.status AS Insurance_case_status
+                        FROM insurance_policy
+                        LEFT JOIN user_profile ON user_profile.id = \
+                        insurance_policy.user
+                        LEFT JOIN crypto_exchange ON crypto_exchange.id = \
+                        insurance_policy.exchange
+                        LEFT JOIN user_payments ON user_payments.id = \
+                        insurance_policy.payment_id
+                        LEFT JOIN insurance_case ON \
+                        insurance_case.insurance = insurance_policy.id
+                        ''')
+    insurance_report = cursor.fetchall()
+
+    if request.method == 'GET':
+        datasets = {}
+        datasets['referral'] = not bool(request.GET.get('referral'))
+        datasets['user'] = not bool(request.GET.get('user'))
+        datasets['exchange'] = not bool(request.GET.get('exchange'))
+        datasets['payments'] = not bool(request.GET.get('payments'))
+        datasets['policy'] = not bool(request.GET.get('policy'))
+        datasets['case'] = not bool(request.GET.get('case'))
+        datasets['additional'] = not bool(request.GET.get('additional'))
+        response = HttpResponse(content_type='application/zip')
+        response['Content-Disposition'] = 'attachment; filename=backup.csv.zip'
+        z = zipfile.ZipFile(response, 'w')
+        for key in datasets:
+            if datasets[key] is True:
+                output = StringIO()
+                writer = csv.writer(output, dialect='excel')
+                query = data[key].objects.all().values()
+                if query.count() > 0:
+                    keys = list(query[0])
+                    writer.writerow(sorted(keys))
+                    for row in query:
+                        writer.writerow([row[k] for k in sorted(keys)])
+                else:
+                    writer.writerow(['NULL TABLE'])
+                z.writestr("%s.csv" % key, output.getvalue())
+
+        out = StringIO()
+        writer = csv.writer(out)
+        header = [
+            'Policy_number', 'Policy_date', 'Name', 'Surname', 'E-mail',
+            'Policy_start_date', 'Policy_expiry_date', 'Number_of_days',
+            'Crypto_exchange_name', 'Limit_BTC',  'Insured_Limit', 'Premium_paid_BTC',
+            'User_paid', 'User_currency', 'Premium_rate_%',
+            'Premium_payment_date', 'Outstanding_claim_BTC', 'Date_of_claim',
+            'Paid_claim_BTC', 'Date_of_claim_payment',
+            'Insurance_policy_status', 'User_payments_status',
+            'Insurance_case_status'
+        ]
+
+        writer.writerow(header)
+        for row in insurance_report:
+            writer.writerow(row)
+        z.writestr("insurance_report.csv", out.getvalue())
+        try:
+            if not z.testzip():
+                responseData = {'error': True, 'message': 'Nothing to backup'}
+                return JsonResponse(responseData)
+        except Exception:
+            return response