bjoern/pypsa-eur
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix retrieve_worldbank_urban_population (#1318)

Browse Source
This commit is contained in:
Fabian Neumann 2024-09-23 12:01:27 +02:00 committed by GitHub
parent e2be4d730e
commit 1a64345067
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
rules/retrieve.smk
View File

@ -346,23 +346,28 @@ if config["enable"]["retrieve"]:
rule retrieve_worldbank_urban_population: rule retrieve_worldbank_urban_population:
params: params:
zip="data/worldbank/API_SP.URB.TOTL.IN.ZS_DS2_en_csv_v2_3403768.zip", zip="data/worldbank/API_SP.URB.TOTL.IN.ZS_DS2_en_csv_v2.zip",
output: output:
gpkg="data/worldbank/API_SP.URB.TOTL.IN.ZS_DS2_en_csv_v2_3403768.csv", gpkg="data/worldbank/API_SP.URB.TOTL.IN.ZS_DS2_en_csv_v2.csv",
run: run:
import os import os
import requests import requests
response = requests.get( response = requests.get(
"https://api.worldbank.org/v2/en/indicator/SP.URB.TOTL.IN.ZS?downloadformat=csv", "https://api.worldbank.org/v2/en/indicator/SP.URB.TOTL.IN.ZS?downloadformat=csv",
params={"name": "API_SP.URB.TOTL.IN.ZS_DS2_en_csv_v2_3403768.zip"},
) )
with open(params["zip"], "wb") as f: with open(params["zip"], "wb") as f:
f.write(response.content) f.write(response.content)
output_folder = Path(params["zip"]).parent output_folder = Path(params["zip"]).parent
unpack_archive(params["zip"], output_folder) unpack_archive(params["zip"], output_folder)
os.remove(params["zip"])
for f in os.listdir(output_folder):
if f.startswith(
"API_SP.URB.TOTL.IN.ZS_DS2_en_csv_v2_"
) and f.endswith(".csv"):
os.rename(os.path.join(output_folder, f), output.gpkg)
break