Fix country set backend

app/api/services/office_country.py

@@ -43,14 +43,21 @@ class OfficeCountryService(BaseService):
     def set_office_country_list(data):
         logger.info('ENTER API::get set_office_country_list')
         try:
-            account_result_data =[]
-            account_result = Country.get_all_countries()
+            val_type = data.get('val_type', '')
+            country_uid = data.get('country_uid', '')
+            code = data.get('code', '')
+            val = data.get('val', '')
+            country_result=[]
+            if val_type == 'STATUS':
+                country_result = Country.set_country_status(country_uid, code, val)
 
+            if val_type == 'SIGNUP':
+                country_result = Country.set_country_signup(country_uid, code, val)
 
 
             account_result = {
-                "set_country_data":[],
-                "message": "Pending implementation"
+                "set_country_data":country_result,
+                "message": "Updated"
             }
             logger.info('RETURN API::get office account view')
             logger.info(account_result)

app/models/country.py

@@ -29,6 +29,23 @@ class Country(db.Model):
             return None
         return country_list
 
+    @classmethod
+    def set_country_status(cls, country_uid, code, val):
+
+        country_data = cls.query.filter_by(uid=str(country_uid), code=str(code)).first()
+        if not country_data:
+            return None
+        country_data.status = val
+        return country_data
+
+    @classmethod
+    def set_country_signup(cls, country_uid, code, val):
+        country_data = cls.query.filter_by(uid=str(country_uid), code=str(code)).first()
+        if not country_data:
+            return None
+        country_data.signup = val
+        return country_data
+
     def to_dict(self):
         return {
             "id": self.id,
@@ -40,7 +57,6 @@ class Country(db.Model):
             "added": self.added.isoformat() if self.added else None
         }
 
-
     '''
      CREATE TABLE country (
 		id SERIAL,