From 6ffd6ec1b82d4b72313f5c89fbba79b355ac9712 Mon Sep 17 00:00:00 2001 From: Michael Tremer Date: Sun, 7 Jan 2024 17:47:19 +0000 Subject: [PATCH] fireinfo: Store ASN Signed-off-by: Michael Tremer --- src/backend/fireinfo.py | 8 +++++--- src/web/fireinfo.py | 3 ++- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/src/backend/fireinfo.py b/src/backend/fireinfo.py index 260572ae..635cd08a 100644 --- a/src/backend/fireinfo.py +++ b/src/backend/fireinfo.py @@ -808,7 +808,7 @@ class Fireinfo(Object): # Handle profile - def handle_profile(self, profile_id, blob, country_code=None, when=None): + def handle_profile(self, profile_id, blob, country_code=None, asn=None, when=None): private_id = blob.get("private_id", None) assert private_id @@ -863,7 +863,8 @@ class Fireinfo(Object): private_id, version, blob, - country_code + country_code, + asn ) VALUES ( @@ -871,9 +872,10 @@ class Fireinfo(Object): %s, %s, %s, + %s, %s ) - """, profile_id, private_id, version, json.dumps(profile), country_code, + """, profile_id, private_id, version, json.dumps(profile), country_code, asn, ) def _validate(self, profile_id, version, blob): diff --git a/src/web/fireinfo.py b/src/web/fireinfo.py index fb33a92c..78fe9e6b 100644 --- a/src/web/fireinfo.py +++ b/src/web/fireinfo.py @@ -37,7 +37,8 @@ class ProfileSendHandler(BaseHandler): with self.db.transaction(): try: self.fireinfo.handle_profile(public_id, profile_blob, - country_code=self.current_country_code) + country_code=self.current_country_code, + asn=self.current_address.asn if self.current_address else None) except ValueError as e: raise tornado.web.HTTPError(400, "Could not process profile: %s" % e) -- 2.47.3