init command

etherdump/commands/init.py

@@ -0,0 +1,119 @@
+from __future__ import print_function
+from argparse import ArgumentParser
+from urlparse import urlparse, urlunparse
+from urllib2 import urlopen, URLError, HTTPError
+from urllib import urlencode
+import json, os, sys
+
+def get_api(url, cmd=None, data=None, verbose=False):
+    try:
+        useurl = url+cmd
+        if data:
+            useurl += "?"+urlencode(data)
+        # data['apikey'] = "7c8faa070c97f83d8f705c935a32d5141f89cbaa2158042fa92e8ddad5dbc5e1"
+        if verbose:
+            print ("trying", useurl, file=sys.stderr)
+        resp = json.load(urlopen(useurl))
+        if "code" in resp and "message" in resp:
+            return resp
+    except ValueError as e:
+        if verbose:
+            print ("  ValueError", e, file=sys.stderr)
+        return
+    except HTTPError as e:
+        if verbose:
+            print ("  HTTPError", e, file=sys.stderr)
+        if e.code == 401:
+            # Unauthorized is how the API responds to an incorrect API key
+            resp = json.load(e)
+            if "code" in resp and "message" in resp:
+                # print ("returning", resp, file=sys.stderr)
+                return resp
+
+def tryapiurl (url, verbose=False):
+    """
+    Try to use url as api, correcting if possible.
+    Returns corrected / normalized URL, or None if not possible
+    """
+    try:
+        scheme, netloc, path, params, query, fragment = urlparse(url)
+        if scheme == "":
+            url = "http://" + url
+        scheme, netloc, path, params, query, fragment = urlparse(url)            
+        params, query, fragment = ("", "", "")
+        path = path.strip("/")
+        # 1. try directly...
+        apiurl = urlunparse((scheme, netloc, path, params, query, fragment))+"/"
+        if get_api(apiurl, "listAllPads", verbose=verbose):
+            return apiurl
+        # 2. try with += api/1.2.9
+        path = os.path.join(path, "api", "1.2.9")+"/"
+        apiurl = urlunparse((scheme, netloc, path, params, query, fragment))
+        if get_api(apiurl, "listAllPads", verbose=verbose):
+            return apiurl
+    # except ValueError as e:
+    #     print ("ValueError", e, file=sys.stderr)
+    except URLError as e:
+        print ("URLError", e, file=sys.stderr)
+
+def main(args):
+    p = ArgumentParser("initialize an etherdump folder")
+    p.add_argument("arg", nargs="*", default=[], help="optional positional args: path etherpadurl")
+    p.add_argument("--path", default=None, help="path to initialize")
+    p.add_argument("--padurl", default=None, help="")
+    p.add_argument("--apikey", default=None, help="")
+    p.add_argument("--verbose", default=False, action="store_true", help="")
+    p.add_argument("--reinit", default=False, action="store_true", help="")
+    args = p.parse_args(args)
+
+
+    path = args.path
+    if path == None and len(args.arg):
+        path = args.arg[0]
+    if not path:
+        path = "."
+
+    edpath = os.path.join(path, ".etherdump")
+    try:
+        os.makedirs(edpath)
+    except OSError:
+        pass
+
+    padinfo = {}
+    padinfopath = os.path.join(edpath, "settings.json")
+    try:
+        with open(padinfopath) as f:
+            padinfo = json.load(f)
+        if not args.reinit:
+            print ("Folder is already initialized. Use --reinit to reset settings.")
+            sys.exit(0)
+    except IOError:
+        pass
+    except ValueError:
+        # bad json file, reset it
+        pass
+
+    apiurl = args.padurl
+    while True:
+        if apiurl:
+            apiurl = tryapiurl(apiurl,verbose=args.verbose)
+        if apiurl:
+            # print ("Got APIURL: {0}".format(apiurl))
+            break
+        apiurl = raw_input("Please type the URL of the etherpad: ").strip()
+    padinfo["apiurl"] = apiurl
+    apikey = args.apikey
+    while True:
+        if apikey:
+            resp = get_api(apiurl, "listAllPads", {"apikey": apikey}, verbose=args.verbose)
+            if resp and resp["code"] == 0:
+                # print ("GOOD")
+                break
+            else:
+                print ("bad")
+        print ("The APIKEY is the contents of the file APIKEY.txt in the etherpad folder", file=sys.stderr)
+        apikey = raw_input("Please paste the APIKEY: ").strip()
+    padinfo["apikey"] = apikey
+
+    with open(padinfopath, "w") as f:
+        json.dump(padinfo, f, indent=2)