Add data version importers for CorePerson, CoreVendor

rattail_corepos/importing/versions.py

@@ -0,0 +1,55 @@
+# -*- coding: utf-8; -*-
+################################################################################
+#
+#  Rattail -- Retail Software Framework
+#  Copyright © 2010-2020 Lance Edgar
+#
+#  This file is part of Rattail.
+#
+#  Rattail is free software: you can redistribute it and/or modify it under the
+#  terms of the GNU General Public License as published by the Free Software
+#  Foundation, either version 3 of the License, or (at your option) any later
+#  version.
+#
+#  Rattail is distributed in the hope that it will be useful, but WITHOUT ANY
+#  WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+#  FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
+#  details.
+#
+#  You should have received a copy of the GNU General Public License along with
+#  Rattail.  If not, see <http://www.gnu.org/licenses/>.
+#
+################################################################################
+"""
+Rattail -> Rattail "versions" data import
+"""
+
+from rattail.importing import versions as base
+
+
+class FromRattailToRattailVersions(base.FromRattailToRattailVersions):
+    """
+    Handler for Rattail -> Rattail "versions" data import
+    """
+
+    def get_importers(self):
+        importers = super(FromRattailToRattailVersions, self).get_importers()
+        importers['CorePerson'] = CorePersonImporter
+        importers['CoreVendor'] = CoreVendorImporter
+        return importers
+
+
+class CorePersonImporter(base.VersionImporter):
+
+    @property
+    def host_model_class(self):
+        model = self.config.get_model()
+        return model.CorePerson
+
+
+class CoreVendorImporter(base.VersionImporter):
+
+    @property
+    def host_model_class(self):
+        model = self.config.get_model()
+        return model.CoreVendor