Skip to content

Commit 2e0665b

Browse files
committed
remove country column
1 parent c01929d commit 2e0665b

File tree

2 files changed

+1
-7
lines changed

2 files changed

+1
-7
lines changed

src/flask_track_usage/__init__.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,6 @@ def after_request(self, response):
172172
'X-Forwarded-For', None),
173173
'authorization': bool(ctx.request.authorization),
174174
'ip_info': None,
175-
'country': None,
176175
'path': ctx.request.path,
177176
'speed': float(speed),
178177
'date': int(time.mktime(current_time.timetuple())),
@@ -195,9 +194,7 @@ def after_request(self, response):
195194
if self.ip_lookup_func:
196195
clean_ip = quote_plus(str(ctx.request.remote_addr))
197196
data['ip_info'] = self.ip_lookup_func(clean_ip)
198-
country = data['ip_info'].get('country', None)
199-
if country:
200-
data['country'] = country
197+
201198
elif self._use_freegeoip:
202199
clean_ip = quote_plus(str(ctx.request.remote_addr))
203200
if '{ip}' in self._freegeoip_endpoint:

src/flask_track_usage/storage/sql.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,6 @@ def set_up(self, engine=None, metadata=None, table_name="flask_usage",
113113
sql.Column('datetime', sql.DateTime),
114114
sql.Column('username', sql.String(128)),
115115
sql.Column('track_var', sql.String(128)),
116-
sql.Column('country', sql.String(64), nullable=True),
117116
)
118117
# Create the table if it does not exist
119118
self.track_table.create(bind=self._eng)
@@ -160,7 +159,6 @@ def store(self, data):
160159
datetime=utcdatetime,
161160
username=data["username"],
162161
track_var=json.dumps(data["track_var"], ensure_ascii=False),
163-
country=data["country"],
164162
)
165163
con.execute(stmt)
166164
return data
@@ -200,7 +198,6 @@ def _get_usage(self, start_date=None, end_date=None, limit=500, page=1):
200198
'date': r[15],
201199
'username': r[16],
202200
'track_var': r[17] if r[17] != '{}' else None,
203-
'country': r[18]
204201
} for r in raw_data]
205202
return usage_data
206203

0 commit comments

Comments
 (0)