Allow multiple CLI arguments

This commit is contained in:
Gabriel Augendre 2019-07-11 22:43:17 +02:00
parent 2ccf67ac6e
commit 24459bc94d

View file

@ -152,13 +152,14 @@ class InseeData:
def main(): def main():
# insee_number = "168127982980507" numbers = ["269059913116714", "168127982980507"]
insee_number = "269059913116714"
if len(sys.argv) > 1: if len(sys.argv) > 1:
insee_number = sys.argv[1] numbers = sys.argv[1:]
data = InseeData(insee_number) for number in numbers:
print(data) data = InseeData(number)
pprint.pprint(data.to_dict()) print(data)
pprint.pprint(data.to_dict())
print('\n\n')
if __name__ == '__main__': if __name__ == '__main__':