debug db.reflect not working

Author: pjamessteven

src/flask_track_usage/storage/sql.py

@@ -107,13 +107,13 @@ def set_up(self, engine=None, metadata=None, table_name="flask_usage",
                     sql.Column('remote_addr', sql.String(24)),
                     sql.Column('xforwardedfor', sql.String(24)),
                     sql.Column('authorization', sql.Boolean),
-                    sql.Column('country', sql.String(64), nullable=True),
                     sql.Column('ip_info', postgresql.JSONB),
                     sql.Column('path', sql.String(128)),
                     sql.Column('speed', sql.Float),
                     sql.Column('datetime', sql.DateTime),
                     sql.Column('username', sql.String(128)),
-                    sql.Column('track_var', sql.String(128))
+                    sql.Column('track_var', sql.String(128)),
+                    sql.Column('country', sql.String(64), nullable=True),
                 )
                 # Create the table if it does not exist
                 self.track_table.create(bind=self._eng)
@@ -154,13 +154,13 @@ def store(self, data):
                 remote_addr=data["remote_addr"],
                 xforwardedfor=data["xforwardedfor"],
                 authorization=data["authorization"],
-                country=data["country"],
                 ip_info=data["ip_info"],
                 path=data["path"],
                 speed=data["speed"],
                 datetime=utcdatetime,
                 username=data["username"],
-                track_var=json.dumps(data["track_var"], ensure_ascii=False)
+                track_var=json.dumps(data["track_var"], ensure_ascii=False),
+                country=data["country"],
             )
             con.execute(stmt)
         return data
@@ -200,6 +200,7 @@ def _get_usage(self, start_date=None, end_date=None, limit=500, page=1):
                 'date': r[15],
                 'username': r[16],
                 'track_var': r[17] if r[17] != '{}' else None
+                'country': r[18]
             } for r in raw_data]
         return usage_data