From e916d4f71f8ace4be5d222a083a7b1903bba5cd1 Mon Sep 17 00:00:00 2001
From: Lance Edgar <ledgar@techsupport.coop>
Date: Tue, 19 Feb 2019 18:11:15 -0600
Subject: [PATCH] Add basic support for editing address for a "contact" record

---
 tailbone/views/customers.py | 22 +++++++++++++++++++++-
 tailbone/views/master.py    | 24 ++++++++++++++++++++++++
 2 files changed, 45 insertions(+), 1 deletion(-)

diff --git a/tailbone/views/customers.py b/tailbone/views/customers.py
index 3651b298..c5cc9fd8 100644
--- a/tailbone/views/customers.py
+++ b/tailbone/views/customers.py
@@ -75,6 +75,11 @@ class CustomersView(MasterView):
         'name',
         'default_phone',
         'default_address',
+        'address_street',
+        'address_street2',
+        'address_city',
+        'address_state',
+        'address_zipcode',
         'default_email',
         'email_preference',
         'wholesale',
@@ -189,12 +194,27 @@ class CustomersView(MasterView):
             f.set_default('default_phone', customer.phones[0].number)
 
         # default_address
-        if self.creating:
+        if self.creating or self.editing:
             f.remove_field('default_address')
         else:
             f.set_renderer('default_address', self.render_default_address)
             f.set_readonly('default_address')
 
+        # address_*
+        if not (self.creating or self.editing):
+            f.remove_fields('address_street',
+                            'address_street2',
+                            'address_city',
+                            'address_state',
+                            'address_zipcode')
+        elif self.editing and customer.addresses:
+            addr = customer.addresses[0]
+            f.set_default('address_street', addr.street)
+            f.set_default('address_street2', addr.street2)
+            f.set_default('address_city', addr.city)
+            f.set_default('address_state', addr.state)
+            f.set_default('address_zipcode', addr.zipcode)
+
         f.set_enum('email_preference', self.enum.EMAIL_PREFERENCE)
         preferences = list(self.enum.EMAIL_PREFERENCE.items())
         preferences.insert(0, ('', "(no preference)"))
diff --git a/tailbone/views/master.py b/tailbone/views/master.py
index 971e344b..97b5063c 100644
--- a/tailbone/views/master.py
+++ b/tailbone/views/master.py
@@ -2731,6 +2731,30 @@ class MasterView(View):
             elif number:
                 contact.add_phone_number(number)
 
+        address_fields = ('address_street',
+                          'address_street2',
+                          'address_city',
+                          'address_state',
+                          'address_zipcode')
+
+        addr = dict([(field, data[field])
+                     for field in address_fields
+                     if field in data])
+
+        if any(addr.values()):
+            # we strip 'address_' prefix from fields
+            addr = dict([(field[8:], value)
+                         for field, value in addr.items()])
+            if contact.addresses:
+                address = contact.addresses[0]
+                for field, value in addr.items():
+                    setattr(address, field, value)
+            else:
+                contact.add_mailing_address(**addr)
+
+        elif contact.addresses:
+            contact.addresses.pop()
+
         return contact
 
     def save_form(self, form):