aboutsummaryrefslogtreecommitdiffstats
path: root/helpers.py
diff options
context:
space:
mode:
Diffstat (limited to 'helpers.py')
-rw-r--r--helpers.py57
1 files changed, 57 insertions, 0 deletions
diff --git a/helpers.py b/helpers.py
new file mode 100644
index 0000000..0e9da70
--- /dev/null
+++ b/helpers.py
@@ -0,0 +1,57 @@
+# Copyright (c) 2018, George Tokmaji
+
+# Permission to use, copy, modify, and/or distribute this software for any
+# purpose with or without fee is hereby granted, provided that the above
+# copyright notice and this permission notice appear in all copies.
+
+# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+
+import sys
+import os, re, json, math
+import requests, hashlib
+from bottle import route, run, Bottle, request, static_file, response, hook, HTTPResponse
+from bson.objectid import ObjectId
+import threading
+
+os.chdir(os.path.dirname(__file__))
+
+io_lock = threading.Lock()
+
+def loadJSON(name : str) -> dict:
+ try:
+ with open(name, "r") as fobj:
+ return json.load(fobj)
+
+ except (FileNotFoundError, json.decoder.JSONDecodeError):
+ return {}
+
+def saveJSON(obj : dict, name : str) -> None:
+ with io_lock:
+ with open(name, "w") as fobj:
+ json.dump(obj, fobj)
+
+database = loadJSON("database.json")
+if "entries" not in database:
+ database["entries"] = {}
+
+def calculateHashForResource(resource : requests.Response, hashobj : object = None) -> object:
+ if hashobj is None:
+ hashobj = hashlib.sha1()
+
+ l = 0
+ for block in resource.iter_content(4096):
+ l += len(block)
+ hashobj.update(block)
+
+ if "content-length" not in resource.headers:
+ resource.headers["Content-Length"] = l
+ return hashobj
+
+def notAllowed():
+ raise HTTPResponse(f"Cannot {request.method} {request.path}")