ad842ee8a5
- Added type hinting (for now, 3.5-compatible) - Split `db` namespace into `db` module and `model` namespace - Changed elastic search to be created lazily for each operation - Changed to class based approach in entity serialization to allow stronger typing - Removed `required` argument from `context.get_*` family of functions; now it's implied if `default` argument is omitted - Changed `unalias_dict` implementation to use less magic inputs
28 lines
938 B
Python
28 lines
938 B
Python
from typing import Any, Optional, List, Dict, Callable
|
|
from szurubooru import db, model, rest, errors
|
|
|
|
|
|
def get_serialization_options(ctx: rest.Context) -> List[str]:
|
|
return ctx.get_param_as_list('fields', default=[])
|
|
|
|
|
|
class BaseSerializer:
|
|
_fields = {} # type: Dict[str, Callable[[model.Base], Any]]
|
|
|
|
def serialize(self, options: List[str]) -> Any:
|
|
field_factories = self._serializers()
|
|
if not options:
|
|
options = list(field_factories.keys())
|
|
ret = {}
|
|
for key in options:
|
|
if key not in field_factories:
|
|
raise errors.ValidationError(
|
|
'Invalid key: %r. Valid keys: %r.' % (
|
|
key, list(sorted(field_factories.keys()))))
|
|
factory = field_factories[key]
|
|
ret[key] = factory()
|
|
return ret
|
|
|
|
def _serializers(self) -> Dict[str, Callable[[], Any]]:
|
|
raise NotImplementedError()
|