From 0186f5f83680a17f67777cadd850d9017a767d63 Mon Sep 17 00:00:00 2001 From: AlexP077 Date: Wed, 22 Mar 2023 19:46:21 +0300 Subject: [PATCH] fix_wkt --- .../migrations/0017_rename_wkt_rivals_wkt.py | 17 +++++++++++++++++ service/models.py | 2 +- service/utils.py | 1 + 3 files changed, 19 insertions(+), 1 deletion(-) create mode 100644 service/migrations/0017_rename_wkt_rivals_wkt.py diff --git a/service/migrations/0017_rename_wkt_rivals_wkt.py b/service/migrations/0017_rename_wkt_rivals_wkt.py new file mode 100644 index 0000000..2c0cb24 --- /dev/null +++ b/service/migrations/0017_rename_wkt_rivals_wkt.py @@ -0,0 +1,17 @@ +# Generated by Django 3.2 on 2023-03-22 16:46 +from django.db import migrations + + +class Migration(migrations.Migration): + + dependencies = [ + ('service', '0016_alter_placementpoint_status'), + ] + + operations = [ + migrations.RenameField( + model_name='rivals', + old_name='WKT', + new_name='wkt', + ), + ] diff --git a/service/models.py b/service/models.py index aa58eb3..1eb3d82 100644 --- a/service/models.py +++ b/service/models.py @@ -54,7 +54,7 @@ class Rayon(models.Model): class Rivals(models.Model): - WKT = gis_models.PointField(srid=SRID, null=True) + wkt = gis_models.PointField(srid=SRID, null=True) info = models.TextField(null=True, blank=True) type = models.TextField(null=True, blank=True) source = models.TextField(null=True, blank=True) diff --git a/service/utils.py b/service/utils.py index b469827..a85df41 100644 --- a/service/utils.py +++ b/service/utils.py @@ -53,6 +53,7 @@ def load_rivals(filepath: str): df = pd.read_csv(filepath) df = df.replace(np.nan, None) df = df.replace('NaT', None) + df.columns = df.columns.str.lower() for row in tqdm(df.to_dict('records'), desc='Loading data...'): models.Rivals.objects.create(**row)