lint: reformat python code with black
This commit is contained in:
parent
331beb01b4
commit
a406a7974b
88 changed files with 2579 additions and 1608 deletions
|
@ -12,7 +12,7 @@ from app.extensions import db
|
|||
|
||||
logger = logging.getLogger(__name__)
|
||||
MAX_DOMAIN_NAME_LENGTH = 255
|
||||
DOMAIN_NAME_REGEX = re.compile(r'^[a-zA-Z0-9.\-]*$')
|
||||
DOMAIN_NAME_REGEX = re.compile(r"^[a-zA-Z0-9.\-]*$")
|
||||
MAX_ALLOWED_ITEMS = 100
|
||||
ListFilter = Union[BinaryExpression[Any], ColumnElement[Any]]
|
||||
|
||||
|
@ -24,7 +24,10 @@ def validate_max_items(max_items_str: str, max_allowed: int) -> int:
|
|||
raise ValueError()
|
||||
return max_items
|
||||
except ValueError:
|
||||
abort(400, description=f"MaxItems must be a positive integer not exceeding {max_allowed}.")
|
||||
abort(
|
||||
400,
|
||||
description=f"MaxItems must be a positive integer not exceeding {max_allowed}.",
|
||||
)
|
||||
|
||||
|
||||
def validate_marker(marker_str: str) -> int:
|
||||
|
@ -47,21 +50,22 @@ TlpMarkings = Union[
|
|||
|
||||
|
||||
def list_resources( # pylint: disable=too-many-arguments,too-many-locals
|
||||
model: Type[Any],
|
||||
serialize_func: Callable[[Any], Dict[str, Any]],
|
||||
*,
|
||||
filters: Optional[List[ListFilter]] = None,
|
||||
order_by: Optional[ColumnElement[Any]] = None,
|
||||
resource_name: str = 'ResourceList',
|
||||
max_items_param: str = 'MaxItems',
|
||||
marker_param: str = 'Marker',
|
||||
max_allowed_items: int = 100,
|
||||
protective_marking: TlpMarkings = 'default',
|
||||
model: Type[Any],
|
||||
serialize_func: Callable[[Any], Dict[str, Any]],
|
||||
*,
|
||||
filters: Optional[List[ListFilter]] = None,
|
||||
order_by: Optional[ColumnElement[Any]] = None,
|
||||
resource_name: str = "ResourceList",
|
||||
max_items_param: str = "MaxItems",
|
||||
marker_param: str = "Marker",
|
||||
max_allowed_items: int = 100,
|
||||
protective_marking: TlpMarkings = "default",
|
||||
) -> ResponseReturnValue:
|
||||
try:
|
||||
marker = request.args.get(marker_param)
|
||||
max_items = validate_max_items(
|
||||
request.args.get(max_items_param, default='100'), max_allowed_items)
|
||||
request.args.get(max_items_param, default="100"), max_allowed_items
|
||||
)
|
||||
query = select(model)
|
||||
|
||||
if filters:
|
||||
|
@ -101,14 +105,21 @@ def list_resources( # pylint: disable=too-many-arguments,too-many-locals
|
|||
abort(500)
|
||||
|
||||
|
||||
def get_single_resource(model: Type[Any], id_: int, resource_name: str) -> ResponseReturnValue:
|
||||
def get_single_resource(
|
||||
model: Type[Any], id_: int, resource_name: str
|
||||
) -> ResponseReturnValue:
|
||||
try:
|
||||
resource = db.session.get(model, id_)
|
||||
if not resource:
|
||||
return jsonify({
|
||||
"Error": "resource_not_found",
|
||||
"Message": f"No {resource_name} found with ID {id_}"
|
||||
}), 404
|
||||
return (
|
||||
jsonify(
|
||||
{
|
||||
"Error": "resource_not_found",
|
||||
"Message": f"No {resource_name} found with ID {id_}",
|
||||
}
|
||||
),
|
||||
404,
|
||||
)
|
||||
return jsonify({resource_name: resource.to_dict()}), 200
|
||||
except Exception: # pylint: disable=broad-exception-caught
|
||||
logger.exception("An unexpected error occurred while retrieving the onion")
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue