1 files changed, 71 insertions, 0 deletions
diff --git a/app/lib/flickr_api/tools.py b/app/lib/flickr_api/tools.py
new file mode 100644
index 0000000..9c9fd68
--- /dev/null
+++ b/app/lib/flickr_api/tools.py
@@ -0,0 +1,71 @@
+from method_call import call_api
+import sys
+import os
+
+
+def load_methods():
+    """
+        Loads the list of all methods
+    """
+    r = call_api(method="flickr.reflection.getMethods")
+    return r["methods"]["method"]
+
+__perms__ = {0: 'none', '1': 'read', '2': 'write', '3': 'delete'}
+
+
+def methods_info():
+    methods = {}
+    for m in load_methods():
+        info = call_api(method="flickr.reflection.getMethodInfo",
+                        method_name=m)
+        info.pop("stat")
+        method = info.pop("method")
+        method["requiredperms"] = __perms__[method["requiredperms"]]
+        method["needslogin"] = bool(method.pop("needslogin"))
+        method["needssigning"] = bool(method.pop("needssigning"))
+        info.update(method)
+        info["arguments"] = info["arguments"]["argument"]
+        info["errors"] = info["errors"]["error"]
+        methods[m] = info
+    return methods
+
+
+def write_reflection(path, template, methods=None):
+    if methods is None:
+        methods = methods_info()
+    with open(template, "r") as t:
+        templ = t.read()
+
+    prefix = ""
+    new_templ = ""
+    tab = "    "
+    templ = templ % str(methods)
+    for c in templ:
+        if c == '{':
+            new_templ += '{\n' + prefix
+            prefix += tab
+        elif c == '}':
+            new_templ += '\n' + prefix + '}\n' + prefix
+            prefix = prefix[:-len(tab)]
+        else:
+            new_templ += c
+
+    with open(path, "w") as f:
+        f.write(new_templ)
+
+
+def write_doc(output_path, exclude=["flickr_keys", "methods"]):
+    import flickr_api
+    exclude.append("__init__")
+    modules = ['flickr_api']
+    dir = os.path.dirname(flickr_api.__file__)
+    modules += [
+        "flickr_api." + f[:-3]
+            for f in os.listdir(dir)
+            if f.endswith(".py") and f[:-3] not in exclude]
+    sys.path.insert(0, dir + "../")
+    if not os.path.exists(output_path):
+        os.makedirs(output_path)
+    os.chdir(output_path)
+    for m in modules:
+        os.system("pydoc -w " + m)