From b6fb53b9a13cccd9e574227f3cac21c35f2f8a8a Mon Sep 17 00:00:00 2001
From: thomaswoehlke <thomas.woehlke@gmail.com>
Date: Tue, 25 May 2021 12:19:30 +0200
Subject: [PATCH] Refactoring: vaccination

---
 .../blueprints/data_divi/divi_model.py        | 33 ++++++++++---------
 .../blueprints/data_who/who_model.py          |  2 +-
 2 files changed, 19 insertions(+), 16 deletions(-)

diff --git a/src/flask_covid19/blueprints/data_divi/divi_model.py b/src/flask_covid19/blueprints/data_divi/divi_model.py
index cfdee989..2f5ae703 100644
--- a/src/flask_covid19/blueprints/data_divi/divi_model.py
+++ b/src/flask_covid19/blueprints/data_divi/divi_model.py
@@ -54,25 +54,28 @@ class DiviCountry(BlueprintLocation):
         db.UniqueConstraint('country_code', 'country', name="uix_divi_country"),
     )
 
+    def __str__(self):
+        return self.location_group.__str__() + " : " + self.location_code + " | " + self.location
+
     id = db.Column(db.Integer, primary_key=True)
-    country_code = db.Column(db.String(255), unique=True, nullable=False)
-    country = db.Column(db.String(255), unique=True, nullable=False)
-    region_id = db.Column(db.Integer, db.ForeignKey('divi_country_region.id'), nullable=False)
-    region = db.relationship(
+    location_code = db.Column(db.String(255), unique=True, nullable=True)
+    location = db.Column(db.String(255), nullable=False, unique=True)
+    location_group_id = db.Column(db.Integer, db.ForeignKey('blueprint_location_group.id'), nullable=False)
+    location_group = db.relationship(
         'DiviRegion',
         lazy='joined',
         cascade='save-update',
-        order_by='DiviRegion.region')
-
-    def __str__(self):
-        result = ""
-        result += self.country_code
-        result += " "
-        result += self.country
-        result += " "
-        result += self.region.region
-        result += " "
-        return result
+        order_by='DiviRegion.location_group')
+    processed_update = db.Column(db.Boolean, nullable=False)
+    processed_full_update = db.Column(db.Boolean, nullable=False)
+    #country_code = db.Column(db.String(255), unique=True, nullable=False)
+    #country = db.Column(db.String(255), unique=True, nullable=False)
+    #region_id = db.Column(db.Integer, db.ForeignKey('divi_country_region.id'), nullable=False)
+    #region = db.relationship(
+    #    'DiviRegion',
+    #    lazy='joined',
+    #    cascade='save-update',
+    #    order_by='DiviRegion.region')
 
     @classmethod
     def remove_all(cls):
diff --git a/src/flask_covid19/blueprints/data_who/who_model.py b/src/flask_covid19/blueprints/data_who/who_model.py
index c434f7f7..2f762b5f 100644
--- a/src/flask_covid19/blueprints/data_who/who_model.py
+++ b/src/flask_covid19/blueprints/data_who/who_model.py
@@ -92,7 +92,7 @@ class WhoCountry(BlueprintLocation):
     processed_full_update = db.Column(db.Boolean, nullable=False)
 
     def __str__(self):
-        return self.location_code + " | " + self.location + " -> " + self.location_group.__str__()
+        return self.location_group.__str__() + " : "+ self.location_code + " | " + self.location
 
     @classmethod
     def find_by_country_code_and_country_and_who_region_id(cls, i_country_code, i_country, my_region):
-- 
GitLab