Changeset 1b69309 in rattail

Timestamp:

07/24/2022 01:21:01 PM (6 months ago)

Author:

Lance Edgar <lance@…>

Branches:

master

Children:

e77ce7f

Parents:

780f781

Message:

Add primary flag for ContactMixin.add_email() and add_phone()

File:

• rattail/db/model/contact.py (modified) (5 diffs)

rattail/db/model/contact.py

@@ -3 +3 @@
 #
 #  Rattail -- Retail Software Framework
-#  Copyright © 2010-2021 Lance Edgar
+#  Copyright © 2010-2022 Lance Edgar
 #
 #  This file is part of Rattail.
@@ -196 +196 @@
         """
         flush = kwargs.pop('flush', True)
+        primary = kwargs.pop('primary', False)
         email = self.make_email(**kwargs)
         self.emails.append(email)
@@ -201 +202 @@
             session = orm.object_session(self)
             session.flush()
+        if primary:
+            self.set_primary_email(email, flush=flush)
         return email
 
@@ -263 +266 @@
         """
         flush = kwargs.pop('flush', True)
+        primary = kwargs.pop('primary', False)
         phone = self.make_phone(**kwargs)
         self.phones.append(phone)
@@ -268 +272 @@
             session = orm.object_session(self)
             session.flush()
+        if primary:
+            self.set_primary_phone(phone, flush=flush)
         return phone