platypush/platypush/backend/http/utils.py
2020-09-27 01:33:38 +02:00

116 lines
3.8 KiB
Python

import json
import logging
import os
import re
from platypush.config import Config
from platypush.backend.http.app import template_folder
class HttpUtils(object):
log = logging.getLogger('platypush:web')
@staticmethod
def widget_columns_to_html_class(columns):
if not isinstance(columns, int):
try:
columns = int(columns)
except ValueError:
raise RuntimeError('columns should be a number, got {} ({})'.format(type(columns), columns))
if 1 <= columns <= 12:
return 'col-{}'.format(columns)
raise RuntimeError('Constraint violation: should be 1 <= columns <= 12, ' +
'got columns={}'.format(columns))
@staticmethod
def search_directory(directory, *extensions, recursive=False):
files = []
if recursive:
for root, subdirs, files in os.walk(directory):
for file in files:
if not extensions or os.path.splitext(file)[1].lower() in extensions:
files.append(os.path.join(root, file))
else:
for file in os.listdir(directory):
if not extensions or os.path.splitext(file)[1].lower() in extensions:
files.append(os.path.join(directory, file))
return files
@classmethod
def search_web_directory(cls, directory, *extensions):
directory = os.path.abspath(os.path.expanduser(directory))
resource_dirs = (Config.get('backend.http') or {}).get('resource_dirs', {})
resource_path = None
uri = ''
for name, resource_path in resource_dirs.items():
resource_path = os.path.abspath(os.path.expanduser(resource_path))
if directory.startswith(resource_path):
subdir = re.sub('^{}(.*)$'.format(resource_path),
'\\1', directory)
uri = '/resources/' + name
break
if not uri:
raise RuntimeError(('Directory {} not found among the available ' +
'static resources on the webserver').format(
directory))
results = [
re.sub('^{}(.*)$'.format(resource_path), uri + '\\1', path)
for path in cls.search_directory(directory, *extensions)
]
return results
@classmethod
def to_json(cls, data):
def json_parse(x):
if type(x) == __import__('datetime').timedelta:
return x.days * 24 * 60 * 60 + x.seconds + x.microseconds / 1e6
# Ignore non-serializable attributes
cls.log.warning('Non-serializable attribute type "{}": {}'.format(type(x), x))
return None
if isinstance(data, type({}.keys())):
# Convert dict_keys to list before serializing
data = list(data)
return json.dumps(data, default=json_parse)
@classmethod
def from_json(cls, data):
return json.loads(data)
@classmethod
def get_config(cls, attr):
return Config.get(attr)
@classmethod
def plugin_name_to_tag(cls, module_name):
return module_name.replace('.','-')
@classmethod
def find_templates_in_dir(cls, directory):
return [
os.path.join(directory, file)
for root, path, files in os.walk(os.path.abspath(os.path.join(template_folder, directory)))
for file in files
if file.endswith('.html') or file.endswith('.htm')
]
@classmethod
def readfile(cls, file):
with open(file) as f:
return f.read()
@classmethod
def isfile(cls, *path):
path = path[0] if len(path) == 1 else os.path.join(*path)
return os.path.isfile(path)
# vim:sw=4:ts=4:et: