Added Apple review export command

1 files changed, 85 insertions, 0 deletions

diff --git a/servo/management/commands/applereview.py b/servo/management/commands/applereview.py
new file mode 100644
index 0000000..4c7ef0b
--- /dev/null
+++ b/servo/management/commands/applereview.py
@@ -0,0 +1,85 @@
+# -*- coding: utf-8 -*-
+
+import logging
+import pyexcel as pe
+from time import time
+from django.core.management.base import BaseCommand
+
+from servo.models import Order
+
+
+class Command(BaseCommand):
+
+    help = "Export data for Apple Review"
+
+    def add_arguments(self, parser):
+        parser.add_argument('year', type=int, help='Start year for report')
+        parser.add_argument('rfp_msg', help='Ready for pickup message')
+        parser.add_argument('status_id', help='ID of "completed" status')
+        parser.add_argument('outfile', help='Path to results file')
+
+    def handle(self, *args, **options):
+        data = []
+        start_time = time()
+        ts_fmt = '%Y-%m-%d'
+        header = [
+            'SERVICE_ORDER', 'DATE_CREATED', 'DATE_COMPLETED',
+            'DATE_CFP',
+            'CUSTOMER_NAME', 'CUSTOMER_ADDRESS', 'CUSTOMER_PHONE',
+            'UNIT_SERIAL_NUMBER', 'REPORTED_SYMPTOMS', 'TECH_SYMPTOMS',
+            'UNIT_WTY_STATUS', 'REPAIR_COST', 'PARTS_REPLACED',
+            'GSX_REPAIRS',
+        ]
+
+        data.append(header)
+        orders = Order.objects.filter(created_at__year__gte=options['year'])
+
+        for o in orders.exclude(customer=None).order_by('created_at'):
+            cust_name = o.get_customer_name() or ''
+            cust_name = cust_name.encode('utf-8')
+            cust_addr = o.customer.street_address
+            cust_phone = o.customer.phone
+            device = o.get_devices().first()
+
+            if device:
+                device_sn = device.sn
+                device_wty = device.get_warranty_status_display()
+            else:
+                device_sn = '-'
+                device_wty = '-'
+
+            notes = o.notes().filter(is_reported=True)
+            if notes:
+                customer_notes = notes[0].body
+                tech_notes = [x.body + ' - ' + x.created_by.full_name for x in notes[1:]]
+            else:
+                customer_notes = ''
+                tech_notes = ''
+
+            parts = [x.code for x in o.get_parts()]
+            repairs = [x.confirmation for x in o.get_repairs()]
+            created_at = o.created_at.strftime(ts_fmt)
+
+            status_closed = o.orderstatus_set.filter(status_id=options['status_id']).first()
+            closed_at = status_closed.started_at.strftime(ts_fmt) if status_closed else '-'
+            rfp_message = o.notes().filter(body__contains=options['rfp_msg']).first()
+
+            if rfp_message:
+                custom_cfp = rfp_message.created_at.strftime(ts_fmt)
+            else:
+                custom_cfp = '-'
+
+            row = [o.code, created_at, closed_at, custom_cfp,
+                   cust_name, cust_addr, cust_phone,
+                   device_sn, customer_notes, ';'.join(tech_notes),
+                   device_wty,
+                   str(o.net_total()),
+                   ', '.join(parts),
+                   ', '.join(repairs)]
+
+            data.append(row)
+
+        sheet = pe.Sheet(data)
+        sheet.save_as(options['outfile'], encoding='UTF-8')
+        seconds = int(time() - start_time)
+        logging.info('%d records exported in %d seconds' % (len(data) - 1), seconds)