forked from Github-Mirrors/canaille
96 lines
2.3 KiB
Python
96 lines
2.3 KiB
Python
import base64
|
|
import hashlib
|
|
import json
|
|
import re
|
|
from base64 import b64encode
|
|
from io import BytesIO
|
|
|
|
from flask import current_app
|
|
from flask import request
|
|
|
|
|
|
def obj_to_b64(obj):
|
|
return base64.b64encode(json.dumps(obj).encode("utf-8")).decode("utf-8")
|
|
|
|
|
|
def b64_to_obj(string):
|
|
return json.loads(base64.b64decode(string.encode("utf-8")).decode("utf-8"))
|
|
|
|
|
|
def build_hash(*args):
|
|
return hashlib.sha256(
|
|
current_app.config["SECRET_KEY"].encode("utf-8")
|
|
+ obj_to_b64(str(args)).encode("utf-8")
|
|
).hexdigest()
|
|
|
|
|
|
def default_fields():
|
|
read = set()
|
|
write = set()
|
|
for acl in current_app.config["CANAILLE"]["ACL"].values():
|
|
if not acl.get("FILTER"):
|
|
read |= set(acl.get("READ", []))
|
|
write |= set(acl.get("WRITE", []))
|
|
|
|
return read, write
|
|
|
|
|
|
def get_current_domain():
|
|
if current_app.config.get("SERVER_NAME"):
|
|
return current_app.config.get("SERVER_NAME")
|
|
|
|
return request.host
|
|
|
|
|
|
def get_current_mail_domain():
|
|
if current_app.config["CANAILLE"]["SMTP"]["FROM_ADDR"]:
|
|
return current_app.config["CANAILLE"]["SMTP"]["FROM_ADDR"].split("@")[-1]
|
|
|
|
return get_current_domain().split(":")[0]
|
|
|
|
|
|
def validate_uri(value):
|
|
regex = re.compile(
|
|
r"^(?:[A-Z0-9\\.-]+)s?://" # scheme + ://
|
|
r"(?:(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+(?:[A-Z]{2,6}\.?|[A-Z0-9-]{2,}\.?)|" # domain...
|
|
r"[A-Z0-9\\.-]+|" # hostname...
|
|
r"\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})" # ...or ip
|
|
r"(?::\d+)?" # optional port
|
|
r"(?:/?|[/?]\S+)$",
|
|
re.IGNORECASE,
|
|
)
|
|
return re.match(regex, value) is not None
|
|
|
|
|
|
class classproperty:
|
|
def __init__(self, f):
|
|
self.f = f
|
|
|
|
def __get__(self, obj, owner):
|
|
return self.f(owner)
|
|
|
|
|
|
def get_b64encoded_qr_image(data):
|
|
try:
|
|
import qrcode
|
|
except ImportError:
|
|
return None
|
|
|
|
qr = qrcode.QRCode(version=1, box_size=10, border=5)
|
|
qr.add_data(data)
|
|
qr.make(fit=True)
|
|
img = qr.make_image(fill_color="black", back_color="white")
|
|
buffered = BytesIO()
|
|
img.save(buffered)
|
|
return b64encode(buffered.getvalue()).decode("utf-8")
|
|
|
|
|
|
def mask_email(email):
|
|
atpos = email.find("@")
|
|
if atpos > 0:
|
|
return email[0] + "#####" + email[atpos - 1 :]
|
|
return None
|
|
|
|
|
|
def mask_phone(phone):
|
|
return phone[0:3] + "#####" + phone[-2:]
|