Using argparse with function, die das ** kwargs-Argument @ annim
Ich benutzeargparse
, um Eingaben zu übernehmen und an eine Funktion zu übergeben, die zwei Variablen und @ als Argumente verwende**kwargs
.
Hier ist meine Funktion:
import requests
import sys
import argparse
def location_by_coordinate(LAT, LNG, **kwargs):
if not kwargs:
coordinate_url = "https://api.instagram.com/v1/locations/search?lat=%s&lng=%s&access_token=%s" % (LAT, LNG, current_token)
r = requests.get(coordinate_url).text
else:
coordinate_url = "https://api.instagram.com/v1/locations/search?lat=%s&lng=%s&access_token=%s" % (LAT, LNG, current_token)
for key, value in kwargs.iteritems():
if 'DISTANCE' in kwargs:
distance = kwargs.get('DISTANCE')
if distance > 5000:
print distance
print "max distance is 5000m, value is reassigned to default of 1000m"
distance = 1000
coordinate_url = "https://api.instagram.com/v1/locations/search?lat=%s&lng=%s&access_token=%s" % (LAT, LNG, current_token)
r = requests.get(coordinate_url).text
else:
pass
coordinate_url = "https://api.instagram.com/v1/locations/search?lat=%s&lng=%s&access_token=%s" % (LAT, LNG, current_token)
r = requests.get(coordinate_url).text
if 'FACEBOOK_PLACES_ID' in kwargs:
fb_places_id = kwargs.get('FACEBOOK_PLACES_ID')
payload = {'FACEBOOK_PLACES_ID': '%s' % (fb_places_id), 'DISTANCE': '%s' % (DISTANCE)}
r = requests.get(coordinate_url, params=payload).text
if 'FOURSQUARE_ID' in kwargs:
foursquare_id = kwargs.get('FOURSQUARE_ID')
payload = {'FOURSQUARE_ID': '%s' % (foursquare_id), 'DISTANCE': '%s' % (DISTANCE)}
r = requests.get(coordinate_url, params=payload).text
if 'FOURSQUARE_V2_ID' in kwargs:
foursquare_v2_id = kwargs.get('FOURSQUARE_V2_ID')
payload = {'FOURSQUARE_V2_ID': '%s' % (foursquare_v2_id), 'DISTANCE': '%s' % (DISTANCE)}
r = requests.get(coordinate_url, params=payload).text
#print r
return r
Bei dieser Funktion und der Verwendung von ** kwargs, wie soll ich die Subparser einrichten?
So habe ich den Kommandozeilen-Parser bisher eingerichtet:
def main():
parser = argparse.ArgumentParser(description="API Endpoints tester")
subparsers = parser.add_subparsers(dest="command", help="Available commands")
location_by_parser = subparsers.add_parser("location_by_coordinate", help="location function")
location_by_parser.add_argument("LAT", help="latitude")
location_by_parser.add_argument("LNG", help="longitude")
arguments = parser.parse_args(sys.argv[1:])
arguments = vars(arguments)
command = arguments.pop("command")
if command == "location_by_coordinate":
LAT, LNG = location_by_coordinate(**arguments)
else:
print "No command provided..."
if __name__ == "__main__":
main()
Offensichtlich funktioniert die obige main () -Funktion gut mit der location_by_coordinate () -Funktion, wenn ich sie in der Befehlszeile wie folgt aufrufe:
$ python argstest.py location_by_coordinate 40.5949799 -73.9495148
Aber mit dem Code, wie er aktuell ist, wenn ich es versuche:
$ python argstest.py location_by_coordinate 40.5949799 -73.9495148 DISTANCE=3000
Offensichtlich bekomme ich:
argstest.py: error: unrecognized arguments: DISTANCE=3000
Aber ich bin mir nicht sicher, wie ich einen Subparser für ** kwargs einrichten soll. Wenn ich versuche, einen Subparser wie folgt einzurichten:
location_by_parser.add_argument("**kwargs", help="**kwargs")
und versuchen Sie den Befehl erneut:
$ python argstest.py location_by_coordinate 40.5949799 -73.9495148 DISTANCE=3000
Das funktioniert nicht, weil dasarguments
Objekt (das ein Wörterbuch ist) wird dies:
{'LAT': '40.5949799', 'LNG': '-73.9495148', 'command': 'location_by_coordinate', '**kwargs': 'DISTANCE=3000'
}
Und dieses Traceback wird zurückgegeben:
Traceback (most recent call last):
File "argstest.py", line 118, in <module>
main()
File "argstest.py", line 108, in main
foo = location_by_coordinate(**arguments)
File "argstest.py", line 40, in location_by_coordinate
return r
UnboundLocalError: local variable 'r' referenced before assignment
Wie kann ich es argparse ermöglichen, die Eingaben in der Befehlszeile zu verarbeiten / zu analysieren, die über ** kwargs an die Funktion übergeben werden sollen?