def main(): args = parser.parse_args() client = ApiClient(response_format=args.format) result = None if args.kvk: result = client.get_by_kvk(kvk=args.kvk) elif args.bedrijfsnaam: result = client.get_by_name(args.bedrijfsnaam) elif args.plaats: result = client.get_by_city(args.plaats, limit=1) else: print("Je hebt geen argumenten ingevoerd") if args.output: with open(args.output, 'w') as o: o.write(result)
def setUp(self): self.client = ApiClient() self.client.setResponseFormat('json')