codeking/sp-hydra-veil-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Implement support for location providers

Browse source
This commit is contained in:
codeking 2025-11-03 22:32:22 +01:00
parent 06769eac62
commit 8c8892bbd4
2 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
core/models/Location.py
View file

@ -12,6 +12,7 @@ _table_definition: str = """
'code' varchar,
'name' varchar,
'time_zone' varchar,
'provider_name' varchar,
UNIQUE(code, country_code)
"""
@ -33,6 +34,10 @@ class Location(Model):
metadata=config(exclude=Exclude.ALWAYS)
)
time_zone: Optional[str] = None
provider_name: Optional[str] = field(
default=None,
metadata=config(exclude=Exclude.ALWAYS)
)
available: Optional[bool] = field(
default=False,
metadata=config(exclude=Exclude.ALWAYS)
@ -71,7 +76,7 @@ class Location(Model):
@staticmethod
def save_many(locations):
Model._create_table_if_not_exists(table_name=_table_name, table_definition=_table_definition)
Model._insert_many('INSERT INTO locations VALUES(?, ?, ?, ?, ?, ?)', Location.tuple_factory, locations)
Model._insert_many('INSERT INTO locations VALUES(?, ?, ?, ?, ?, ?, ?)', Location.tuple_factory, locations)
@staticmethod
def factory(cursor, row):
@ -80,4 +85,4 @@ class Location(Model):
@staticmethod
def tuple_factory(location):
return location.id, location.country_code, location.country_name, location.code, location.name, location.time_zone
return location.id, location.country_code, location.country_name, location.code, location.name, location.time_zone, location.provider_name

2
core/services/WebServiceApiService.py
View file

@ -59,7 +59,7 @@ class WebServiceApiService:
if response.status_code == requests.codes.ok:
for location in response.json()['data']:
locations.append(Location(location['country']['code'], location['code'], location['id'], location['country']['name'], location['name'], location['time_zone']['code']))
locations.append(Location(location['country']['code'], location['code'], location['id'], location['country']['name'], location['name'], location['time_zone']['code'], location['provider']['name']))
return locations