mirror of
https://github.com/lucaspalomodevelop/eventcally.git
synced 2026-03-13 00:07:22 +00:00
59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
import os
|
|
import pathlib
|
|
|
|
from flask_babelex import lazy_gettext
|
|
from psycopg2.errorcodes import CHECK_VIOLATION, UNIQUE_VIOLATION
|
|
from sqlalchemy.exc import IntegrityError
|
|
|
|
|
|
def get_event_category_name(category):
|
|
return lazy_gettext("Event_" + category.name)
|
|
|
|
|
|
def get_localized_enum_name(enum):
|
|
return lazy_gettext(enum.__class__.__name__ + "." + enum.name)
|
|
|
|
|
|
def get_localized_scope(scope: str) -> str:
|
|
loc_key = "Scope_" + scope
|
|
return lazy_gettext(loc_key)
|
|
|
|
|
|
def make_dir(path):
|
|
try:
|
|
original_umask = os.umask(0)
|
|
pathlib.Path(path).mkdir(parents=True, exist_ok=True)
|
|
finally:
|
|
os.umask(original_umask)
|
|
|
|
|
|
def clear_files_in_dir(path):
|
|
with os.scandir(path) as entries:
|
|
for entry in entries:
|
|
if entry.is_file() or entry.is_symlink():
|
|
os.remove(entry.path)
|
|
|
|
|
|
def split_by_crlf(s):
|
|
return [v for v in s.splitlines() if v]
|
|
|
|
|
|
def make_integrity_error(
|
|
pgcode: str, message: str = "", statement: str = None
|
|
) -> IntegrityError:
|
|
class Psycog2Error(object):
|
|
def __init__(self, pgcode, message):
|
|
self.pgcode = pgcode
|
|
self.message = message
|
|
|
|
orig = Psycog2Error(pgcode, message)
|
|
return IntegrityError(statement, list(), orig)
|
|
|
|
|
|
def make_check_violation(message: str = None, statement: str = "") -> IntegrityError:
|
|
return make_integrity_error(CHECK_VIOLATION, message, statement)
|
|
|
|
|
|
def make_unique_violation(message: str = None, statement: str = "") -> IntegrityError:
|
|
return make_integrity_error(UNIQUE_VIOLATION, message, statement)
|