use 2.0 joblib

1 files changed, 40 insertions, 0 deletions

diff --git a/cli.py b/cli.py
new file mode 100644
index 0000000..f26901a
--- /dev/null
+++ b/cli.py
@@ -0,0 +1,40 @@
+#!/usr/bin/python3
+import os
+import argparse
+
+import joblib.joblib
+
+
+RABBIT = {
+    "host": os.environ.get("RABBITMQ_HOST"),
+    "port": os.environ.get("RABBITMQ_PORT"),
+    "credentials": os.environ.get("RABBITMQ_CREDS")
+}
+
+
+def main(args):
+    queue = joblib.joblib.JobQueue()
+    queue.connect(**RABBIT)
+
+    if args.player:
+        payload = {
+            "region": args.region,
+            "params": {
+                "filter[createdAt-start]": "2017-02-01T00:00:00Z",
+                "filter[playerNames]": args.player
+            }
+        }
+        queue.request("grab", payload)
+
+parser = argparse.ArgumentParser(description="Request a Vainsocial update.")
+parser.add_argument("-n", "--player",
+                    help="Player name",
+                    type=str)
+parser.add_argument("-r", "--region",
+                    help="Specify a region",
+                    type=str,
+                    choices=["na", "eu", "sg"],
+                    default="na")
+
+if __name__ == "__main__":
+    main(parser.parse_args())