Added translation code for other models

Author: shi-jie-samuel-tan

database/bupehandler/model_translation.py

@@ -1,4 +1,4 @@
-class sites_general_display(): 
+class Sites_general_display: 
     def __init__(self, table_name, source_object): 
         self.table_name = table_name
         self.output = {"name1": None, "name2": None, "name3": None, "website1": None, 
@@ -16,11 +16,29 @@ def __init__(self, table_name, source_object):
             "intake1" : "phone2", 
             "intake2" : "phone3", 
             }
-            for key in source_object: 
-                if key in self.output: 
-                    self.output[key] = source_object[key] 
-                elif key in mapping: 
-                    self.output[mapping[key]] = source_object[key] 
+        elif self.table_name == "sites_all": 
+            mapping = dict()
+        elif self.table_name == "siterecs_samhsa_otp": 
+            mapping = {"program_name": "name1", 
+            "dba": "name2", 
+            "phone": "phone1", 
+            "street" : "street1" 
+            }
+        elif self.table_name == "siterecs_dbhids_tad": 
+            mapping = {"phone": "phone1"}
+        elif self.table_name == "siterecs_hfp_fqhc": 
+            mapping = {"name_system": "name1",
+            "name_site": "name2",
+            "name_short": "name3",
+            "website": "website1"
+            }
+        elif self.table_name == "siterecs_other_srcs": 
+            mapping = dict()
+        for key in source_object: 
+            if key in self.output: 
+                self.output[key] = source_object[key] 
+            elif key in mapping: 
+                self.output[mapping[key]] = source_object[key] 
 
 
 

database/bupehandler/views.py

@@ -14,7 +14,7 @@
 from .models import Sitecodes_samhsa_ftloc, Siterecs_samhsa_ftloc, Siterecs_samhsa_otp, Siterecs_dbhids_tad, Siterecs_other_srcs, Sites_all
 import re 
 from spellchecker import SpellChecker
-from .model_translation import sites_general_display
+from .model_translation import Sites_general_display
 from django.forms.models import model_to_dict
 @api_view(["GET", "POST", "DELETE"])
 @csrf_exempt
@@ -123,7 +123,7 @@ def filtered_table(request, table_name, param_values=None, excluded_values=None)
         table_objects = table_objects.order_by(*order_by_list)
     general_display_list = [] 
     for table_object in table_objects: 
-        general_display_list.append(sites_general_display(table_name, model_to_dict(table_object)).output)
+        general_display_list.append(Sites_general_display(table_name, model_to_dict(table_object)).output)
     #table_serializer = serializer_dict[table_name](table_objects, many=True)
     print(len(general_display_list))
     return render(request,"bupehandler/list_all.html", {"title": table_name, "objects" : general_display_list})