90 lines
1.9 KiB
Python
90 lines
1.9 KiB
Python
from datetime import datetime
|
|
from math import ceil
|
|
import os
|
|
from os.path import isfile
|
|
|
|
def list_files(path, recursive=False):
|
|
if recursive:
|
|
for p, dirnames, fils in os.walk(path):
|
|
for f in fils:
|
|
fil = os.path.join(p, f)
|
|
yield fil
|
|
else:
|
|
for f in os.listdir(path):
|
|
fil = os.path.join(path, f)
|
|
if not isfile(fil):
|
|
continue
|
|
yield fil
|
|
|
|
def must_get(d, k):
|
|
if type(k) == str:
|
|
k = k.split('.')
|
|
part = k.pop(0)
|
|
if type(d) != dict or part not in d:
|
|
raise ValueError(part + ' not found')
|
|
val = d[part]
|
|
if len(k) == 0:
|
|
return val
|
|
else:
|
|
return must_get(val, k)
|
|
|
|
mg = must_get
|
|
|
|
def should_get(d, k, default=None):
|
|
try:
|
|
return must_get(d, k)
|
|
except ValueError:
|
|
return default
|
|
|
|
sg = should_get
|
|
|
|
def find_where_eq(l, k, v):
|
|
for i in l:
|
|
if k in i and i[k] == v:
|
|
return i
|
|
return None
|
|
|
|
def all_subclasses(cls):
|
|
return set(cls.__subclasses__()).union(
|
|
[s for c in cls.__subclasses__() for s in all_subclasses(c)])
|
|
|
|
def pprint(d, detail=2):
|
|
print(pretty(d, detail=detail))
|
|
|
|
def pretty(d, ident=0, detail=2):
|
|
if type(d) in [int, str, float, bool]:
|
|
return str(d)
|
|
if hasattr(d, 'f'):
|
|
return d.f(detail)
|
|
r = ''
|
|
idt = ' ' * ident
|
|
if type(d) == list:
|
|
r += 'lst'
|
|
for i in d:
|
|
r += '\n' + idt + pretty(i, ident + 1, detail)
|
|
elif type(d) == dict:
|
|
r += 'map'
|
|
for k,v in d.items():
|
|
r += '\n' + idt + k + ': ' + pretty(v, ident + 1, detail)
|
|
return r
|
|
|
|
def trim(s, l):
|
|
s = s[:l]
|
|
s += ' ' * (l-len(s))
|
|
return s
|
|
|
|
def total_pages(meta):
|
|
total = mg(meta, 'total')
|
|
limit = mg(meta, 'limit')
|
|
pages = ceil(total / limit)
|
|
return pages
|
|
|
|
# >>> parse_timestamp('2023-06-02T20:34:48.293Z')
|
|
# 1685738088
|
|
def parse_timestamp(ts):
|
|
return int(datetime.strptime(ts, '%Y-%m-%dT%H:%M:%S.%f%z').timestamp())
|
|
|
|
def render_timestamp(ts):
|
|
return datetime.utcfromtimestamp(ts).isoformat()
|
|
|