Browse Source

init command

add-quote-import
Michael Murtaugh 9 years ago
parent
commit
59f2b7717c
  1. 119
      etherdump/commands/init.py

119
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)
Loading…
Cancel
Save