Skip to content

Commit

Permalink
chore: remove useless use of OOP
Browse files Browse the repository at this point in the history
  • Loading branch information
@zeha
zeha committed Jun 9, 2020
1 parent be08904 commit 7d13ed6
Showing 1 changed file with 126 additions and 131 deletions.
257 changes: 126 additions & 131 deletions grml-paste
View file
Original file line number Diff line number Diff line change
Expand Up @@ -30,120 +30,113 @@ class ActionFailedException(Exception):
return self.args[1]


class Action(object):
def __init__(self, args, opts):
self.args_ = args
self.opts_ = opts

def _createProxy(self):
return ServerProxy(self.opts_.server, verbose=False)

def _callProxy(self, functor, server=None):
"""Wrapper for xml-rpc calls to server which throws an
ActionFailedException on error"""
if server is None:
server = self._createProxy()
ret = functor(server)
if ret["rc"] != 0:
raise ActionFailedException(ret["statusmessage"], ret)
return ret

def call(self, method_name):
"""External Interface to call the appropriate action"""
return self.__getattribute__(method_name)()

def actionAddPaste(self):
"""Add paste to the server: <1.line> <2.line> ...
default Read paste from stdin.
[text] Every argument on the commandline will be interpreted as
a seperate line of paste.
"""
server = self._createProxy()
o = self.opts_
code = self.args_
if len(self.args_) == 0:
code = [i.rstrip() for i in sys.stdin.readlines()]
code = "\n".join(code)
result = self._callProxy(lambda s: s.paste.addPaste(code, o.name, o.expire * 3600, o.lang, o.private),
server)
return result["statusmessage"], result

def actionDelPaste(self):
"""Delete paste from server: <digest>
<digest> Digest of paste you want to remove.
"""
digest = self.args_.pop(0)
result = self._callProxy(lambda s: s.paste.deletePaste(digest))
return result["statusmessage"], result

def actionGetPaste(self):
"""Get paste from server: <id>
<id> Id of paste you want to receive.
"""
paste_id = self.args_.pop(0)
result = self._callProxy(lambda s: s.paste.getPaste(paste_id))
return result["code"], result

def actionGetLangs(self):
"""Get supported language highlighting types from server"""
result = self._callProxy(lambda s: s.paste.getLanguages())
return "\n".join(result["langs"]), result

def actionAddShortUrl(self):
"""Add short-URL: <url>
<url> Short-URL to add
"""
url = self.args_.pop(0)
result = self._callProxy(lambda s: s.paste.addShortURL(url))
return result["url"], result

def actionGetShortUrl(self):
"""Resolve short-URL: <url>
<url> Short-URL to get clicks of
"""
url = self.args_.pop(0)
result = self._callProxy(lambda s: s.paste.resolveShortURL(url))
return (result["url"], result)

def actionGetShortUrlClicks(self):
"""Get clicks of short-URL: <url>
<url> Short-URL to get clicks of
"""
url = self.args_.pop(0)
result = self._callProxy(lambda s: s.paste.ShortURLClicks(url))
return result["count"], result

def actionHelp(self):
"""Print more verbose help about specific action: <action>
<action> Topic on which you need more verbose help.
"""
if len(self.args_) < 1:
alias = "help"
else:
alias = self.args_.pop(0)

if alias in actions:
fun = actions[alias]
print(inspect.getdoc(self.__getattribute__(fun)))
print("\naliase: " + " ".join([i for i in actions_r[fun] if i != alias]))
else:
print("Error: No such command - %s" % (alias))
OPT_PARSER.print_usage()
sys.exit(0)


# actionAddPaste -> [add, a]
def _paste(opts):
"""Get paste proxy object"""
return (ServerProxy(opts.server, verbose=False).paste)


def _check_success(return_data):
"""Check if call was successful, raise AcitonFailedException otherwise"""
if return_data["rc"] != 0:
raise ActionFailedException(return_data["statusmessage"], return_data)
return return_data


def action_add_paste(opts, code):
"""Add paste to the server: <1.line> <2.line> ...
default Read paste from stdin.
[text] Every argument on the commandline will be interpreted as
a seperate line of paste.
"""
if len(code) == 0:
code = [line.rstrip() for line in sys.stdin.readlines()]
code = "\n".join(code)
result = _check_success(_paste(opts).addPaste(code, opts.name, opts.expire * 3600, opts.lang, opts.private))
return result["statusmessage"], result


def action_del_paste(opts, args):
"""Delete paste from server: <digest>
<digest> Digest of paste you want to remove.
"""
digest = args.pop(0)
result = _check_success(_paste(opts).deletePaste(digest))
return result["statusmessage"], result


def action_get_paste(opts, args):
"""Get paste from server: <id>
<id> Id of paste you want to receive.
"""
paste_id = args.pop(0)
result = _check_success(_paste(opts).getPaste(paste_id))
return result["code"], result


def action_get_langs(opts, args):
"""Get supported language highlighting types from server"""
result = _check_success(_paste(opts).getLanguages())
return "\n".join(result["langs"]), result


def action_add_short_url(opts, args):
"""Add short-URL: <url>
<url> Short-URL to add
"""
url = args.pop(0)
result = _check_success(_paste(opts).addShortURL(url))
return result["url"], result


def action_get_short_url(opts, args):
"""Resolve short-URL: <url>
<url> Short-URL to get clicks of
"""
url = args.pop(0)
result = _check_success(_paste(opts).resolveShortURL(url))
return result["url"], result


def action_get_short_url_clicks(opts, args):
"""Get clicks of short-URL: <url>
<url> Short-URL to get clicks of
"""
url = args.pop(0)
result = _check_success(_paste(opts).ShortURLClicks(url))
return result["count"], result


def action_help(opts, args):
"""Print more verbose help about specific action: <action>
<action> Topic on which you need more verbose help.
"""
if len(args) < 1:
alias = "help"
else:
alias = args.pop(0)

if alias in actions:
function_name = actions[alias]
print(inspect.getdoc(globals()[function_name]))
print("\naliases: " + " ".join([i for i in actions_r[function_name] if i != alias]))
else:
print("Error: No such command - %s" % (alias))
OPT_PARSER.print_usage()
sys.exit(0)


# action_add_paste -> [add, a]
actions_r = {}

# add -> actionAddPaste
# a -> actionAddPaste
# add -> action_add_paste
# a -> action_add_paste
actions = {}

# option parser
Expand All @@ -154,26 +147,28 @@ OPT_PARSER = None
# MAIN
##
if __name__ == "__main__":
action_spec = ["actionAddPaste add a",
"actionDelPaste del d rm",
"actionGetPaste get g",
"actionGetLangs getlangs gl langs l",
"actionAddShortUrl addurl",
"actionGetShortUrl geturl",
"actionGetShortUrlClicks getclicks",
"actionHelp help"]
for i in action_spec:
aliases = i.split()
action_specs = [
"action_add_paste add a",
"action_del_paste del d rm",
"action_get_paste get g",
"action_get_langs getlangs gl langs l",
"action_add_short_url addurl",
"action_get_short_url geturl",
"action_get_short_url_clicks getclicks",
"action_help help"
]
for action_spec in action_specs:
aliases = action_spec.split()
cmd = aliases.pop(0)
actions_r[cmd] = aliases
for (k, v) in actions_r.items():
for (action_name, v) in actions_r.items():
for i in v:
actions[i] = k
actions[i] = action_name

usage = "usage: %prog [options] ACTION <args>\n\n" +\
"actions:\n" +\
"\n".join(["%12s\t%s" % (v[0], inspect.getdoc(getattr(Action, k)).split("\n")[0])
for (k, v) in actions_r.items()])
"\n".join(["%12s\t%s" % (v[0], inspect.getdoc(globals()[action_name]).splitlines()[0])
for (action_name, v) in actions_r.items()])
running_user = getpass.getuser()
parser = optparse.OptionParser(usage=usage)
parser.add_option("-n", "--name", default=running_user, help="Name of poster")
Expand All @@ -192,17 +187,17 @@ if __name__ == "__main__":
parser.error("Please provide me with an action")
elif args[0] in actions:
cmd = args.pop(0)
action = Action(args, opts)
action = actions[cmd]
try:
(msg, ret) = action.call(actions[cmd])
(msg, ret) = globals()[action](opts, args)
if opts.verbose == 0:
print(msg)
else:
print(ret)
except ActionFailedException as e:
sys.stderr.write("Server Error: %s\n" % e.what())
except ActionFailedException as except_inst:
sys.stderr.write("Server Error: %s\n" % except_inst.what())
if opts.verbose > 0:
print(e.dwhat())
print(except_inst.dwhat())
sys.exit(1)
else:
parser.error("Unknown action: %s" % args[0])

0 comments on commit 7d13ed6

Please sign in to comment.