import re import os import hashlib import json from flask import request from recaptcha.client import captcha from realms import config from realms.lib.services import cache def cache_it(fn): def wrap(*args, **kw): key = "%s:%s" % (args[0].table, args[1]) data = cache.get(key) # Assume strings are JSON encoded try: data = json.loads(data) except TypeError: pass except ValueError: pass if data is not None: return data else: data = fn(*args) print data ret = data if data is None: data = '' if not isinstance(data, basestring): try: data = json.dumps(data, separators=(',', ':')) except TypeError: pass cache.set(key, data) return ret return wrap def to_json(res, first=False): """ Jsonify query result. """ res = to_dict(res, first) return json.dumps(res, separators=(',', ':')) def to_dict(cur, first=False): if not cur: return None ret = [] for row in cur: ret.append(row) if ret and first: return ret[0] else: return ret def validate_captcha(): response = captcha.submit( request.form['recaptcha_challenge_field'], request.form['recaptcha_response_field'], config.flask['RECAPTCHA_PRIVATE_KEY'], request.remote_addr) return response.is_valid def mkdir_safe(path): if path and not(os.path.exists(path)): os.makedirs(path) return path def extract_path(file_path): if not file_path: return None last_slash = file_path.rindex("/") if last_slash: return file_path[0, last_slash] def clean_path(path): if path: if path[0] != '/': path.insert(0, '/') return re.sub(r"//+", '/', path) def extract_name(file_path): if file_path[-1] == "/": return None return os.path.basename(file_path) def remove_ext(path): return os.path.splitext(path)[0] def clean_url(url): if not url: return url url = url.replace('%2F', '/') url = re.sub(r"^/+", "", url) return re.sub(r"//+", '/', url) def to_canonical(s): """ Double space -> single dash Double dash -> single dash Remove all non alphanumeric and dash Limit to first 64 chars """ s = s.encode('ascii', 'ignore') s = str(s) s = re.sub(r"\s\s*", "-", s) s = re.sub(r"\-\-+", "-", s) s = re.sub(r"[^a-zA-Z0-9\-]", "", s) s = s[:64] return s def gravatar_url(email): return "https://www.gravatar.com/avatar/" + hashlib.md5(email).hexdigest()