Merge pull request #81 from gm3dmo/ranks

Adding Ranks model, deleter and loader for #80

cmp/models.py

@@ -16,6 +16,15 @@ class CustomUser(AbstractUser):
     def __str__(self):
         return self.email
 
+class Rank(models.Model):
+    RankTypes = (('OR','Other Rank'),('NC','Non Commisioned Officer'),('OF','Officer'))
+    Name = models.CharField(max_length=50, unique=True)
+    Abbreviation = models.CharField(max_length=50, blank=True)
+    Class = models.CharField(max_length=2, blank=True, choices=RankTypes,default='Other Rank')
+
+    def __str__(self):
+        return self.Name
+
 
 class Country(models.Model):
     name_common = models.CharField(max_length=255, unique=False, default='')

scripts/delete-all-ranks.py

@@ -0,0 +1,5 @@
+from cmp.models import Rank
+
+def run():
+    Ranks = Rank.objects.all()
+    Ranks.delete()

scripts/insert-all-ranks.py

@@ -0,0 +1,38 @@
+
+def run():
+
+    import urllib3
+    import csv
+    from cmp.models import Rank
+
+    ref_data_url = "https://raw.githubusercontent.com/gm3dmo/old-cmp/main/data/ranks.csv"
+    http = urllib3.PoolManager()
+    r = http.request('GET', ref_data_url)
+    print(r.status)
+    # load the response into a csv dictionary reader
+    reader = csv.DictReader(r.data.decode('utf-8').splitlines())
+
+    # add a country model for each row in the csv file
+    for row in reader:
+        print(row['name'])
+        try:
+            Rank.objects.create(
+                id=row['id'],
+                Name=row['name'],
+                Abbreviation=row['abbr'],
+                Class=row['class']
+        )
+        except Exception as e:
+            print("Error with: " + row['name'])
+            raise e
+
+    for rank in Rank.objects.all():
+       print(f"""{rank.id} {rank.Name}""")
+
+
+
+
+
+
+
+