mirror of
https://github.com/morpheus65535/bazarr.git
synced 2024-11-15 20:24:42 +08:00
129 lines
4.5 KiB
Python
129 lines
4.5 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: UTF-8 -*-
|
|
|
|
# Abstract converter functions for use in any Box class
|
|
|
|
import csv
|
|
import json
|
|
import sys
|
|
import warnings
|
|
from pathlib import Path
|
|
|
|
import dynaconf.vendor.ruamel.yaml as yaml
|
|
from dynaconf.vendor.box.exceptions import BoxError, BoxWarning
|
|
from dynaconf.vendor import tomllib as toml
|
|
|
|
|
|
BOX_PARAMETERS = ('default_box', 'default_box_attr', 'conversion_box',
|
|
'frozen_box', 'camel_killer_box',
|
|
'box_safe_prefix', 'box_duplicates', 'ordered_box',
|
|
'default_box_none_transform', 'box_dots', 'modify_tuples_box',
|
|
'box_intact_types', 'box_recast')
|
|
|
|
|
|
def _exists(filename, create=False):
|
|
path = Path(filename)
|
|
if create:
|
|
try:
|
|
path.touch(exist_ok=True)
|
|
except OSError as err:
|
|
raise BoxError(f'Could not create file {filename} - {err}')
|
|
else:
|
|
return
|
|
if not path.exists():
|
|
raise BoxError(f'File "{filename}" does not exist')
|
|
if not path.is_file():
|
|
raise BoxError(f'{filename} is not a file')
|
|
|
|
|
|
def _to_json(obj, filename=None, encoding="utf-8", errors="strict", **json_kwargs):
|
|
json_dump = json.dumps(obj, ensure_ascii=False, **json_kwargs)
|
|
if filename:
|
|
_exists(filename, create=True)
|
|
with open(filename, 'w', encoding=encoding, errors=errors) as f:
|
|
f.write(json_dump if sys.version_info >= (3, 0) else json_dump.decode("utf-8"))
|
|
else:
|
|
return json_dump
|
|
|
|
|
|
def _from_json(json_string=None, filename=None, encoding="utf-8", errors="strict", multiline=False, **kwargs):
|
|
if filename:
|
|
_exists(filename)
|
|
with open(filename, 'r', encoding=encoding, errors=errors) as f:
|
|
if multiline:
|
|
data = [json.loads(line.strip(), **kwargs) for line in f
|
|
if line.strip() and not line.strip().startswith("#")]
|
|
else:
|
|
data = json.load(f, **kwargs)
|
|
elif json_string:
|
|
data = json.loads(json_string, **kwargs)
|
|
else:
|
|
raise BoxError('from_json requires a string or filename')
|
|
return data
|
|
|
|
|
|
def _to_yaml(obj, filename=None, default_flow_style=False, encoding="utf-8", errors="strict", **yaml_kwargs):
|
|
if filename:
|
|
_exists(filename, create=True)
|
|
with open(filename, 'w',
|
|
encoding=encoding, errors=errors) as f:
|
|
yaml.dump(obj, stream=f, default_flow_style=default_flow_style, **yaml_kwargs)
|
|
else:
|
|
return yaml.dump(obj, default_flow_style=default_flow_style, **yaml_kwargs)
|
|
|
|
|
|
def _from_yaml(yaml_string=None, filename=None, encoding="utf-8", errors="strict", **kwargs):
|
|
if 'Loader' not in kwargs:
|
|
kwargs['Loader'] = yaml.SafeLoader
|
|
if filename:
|
|
_exists(filename)
|
|
with open(filename, 'r', encoding=encoding, errors=errors) as f:
|
|
data = yaml.load(f, **kwargs)
|
|
elif yaml_string:
|
|
data = yaml.load(yaml_string, **kwargs)
|
|
else:
|
|
raise BoxError('from_yaml requires a string or filename')
|
|
return data
|
|
|
|
|
|
def _to_toml(obj, filename=None, encoding="utf-8", errors="strict"):
|
|
if filename:
|
|
_exists(filename, create=True)
|
|
with open(filename, 'w', encoding=encoding, errors=errors) as f:
|
|
toml.dump(obj, f)
|
|
else:
|
|
return toml.dumps(obj)
|
|
|
|
|
|
def _from_toml(toml_string=None, filename=None, encoding="utf-8", errors="strict"):
|
|
if filename:
|
|
_exists(filename)
|
|
with open(filename, 'r', encoding=encoding, errors=errors) as f:
|
|
data = toml.load(f)
|
|
elif toml_string:
|
|
data = toml.loads(toml_string)
|
|
else:
|
|
raise BoxError('from_toml requires a string or filename')
|
|
return data
|
|
|
|
|
|
def _to_csv(box_list, filename, encoding="utf-8", errors="strict"):
|
|
csv_column_names = list(box_list[0].keys())
|
|
for row in box_list:
|
|
if list(row.keys()) != csv_column_names:
|
|
raise BoxError('BoxList must contain the same dictionary structure for every item to convert to csv')
|
|
|
|
if filename:
|
|
_exists(filename, create=True)
|
|
with open(filename, 'w', encoding=encoding, errors=errors, newline='') as csv_file:
|
|
writer = csv.DictWriter(csv_file, fieldnames=csv_column_names)
|
|
writer.writeheader()
|
|
for data in box_list:
|
|
writer.writerow(data)
|
|
|
|
|
|
def _from_csv(filename, encoding="utf-8", errors="strict"):
|
|
_exists(filename)
|
|
with open(filename, 'r', encoding=encoding, errors=errors, newline='') as f:
|
|
reader = csv.DictReader(f)
|
|
return [row for row in reader]
|