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?

Antworten auf die Frage(4)

Ihre Antwort auf die Frage