kbruen/cfr-xml-to-sql
1
0
Fork 0
Code Issues Pull requests Releases Activity

Using tqdm description instead of printing

Browse source
This commit is contained in:
Kenneth Bruen 2022-01-23 23:20:00 +02:00
parent f1724eb90c
commit 34dc772aea
Signed by: kbruen
GPG key ID: C1980A470C3EE5B1
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
convert.py
View file

@ -4,9 +4,11 @@ import xml.etree.ElementTree as ET
from itertools import takewhile
try:
from tqdm import tqdm
TQDM_IMPORTED = True
except ImportError:
def tqdm(iter, *args, **kargs):
return iter
TQDM_IMPORTED = False
def get_database_connection():
import sqlite3
@ -158,8 +160,9 @@ def main():
trains = find_trains(con)
print(f'Adding {company_common_name or f}...')
for el_tren in tqdm(el_trenuri.findall("./Tren")):
if not TQDM_IMPORTED:
print(f'Adding {company_common_name or f}...')
for el_tren in tqdm(el_trenuri.findall("./Tren"), desc=company_common_name or f):
train_number_str = el_tren.attrib['Numar']
train_number = train_number_stoi(train_number_str)