1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
"""Functions to perform common checks.
These are useful for reusability, and hence maintainability of the code.
"""
import logging
from typing import Callable
from functools import wraps
from gn_libs.mysqldb import Connection, database_connection
from flask import flash, url_for, redirect, current_app as app
from uploader.species.models import species_by_id
from uploader.population.models import population_by_species_and_id
logger = logging.getLogger(__name__)
def with_species(redirect_uri: str):
"""Ensure the species actually exists."""
def __decorator__(function):
@wraps(function)
def __with_species__(**kwargs):
try:
species_id = int(kwargs.get("species_id"))
if not bool(species_id):
flash("Expected species_id value to be present!",
"alert-danger")
return redirect(url_for(redirect_uri))
with database_connection(app.config["SQL_URI"]) as conn:
species = species_by_id(conn, species_id)
if not bool(species):
flash("Could not find species with that ID",
"alert-danger")
return redirect(url_for(redirect_uri))
except ValueError as _verr:
logger.debug(
"Exception converting value to integer: %s",
kwargs.get("species_id"),
exc_info=True)
flash("Expected an integer for 'species_id' value.",
"alert-danger")
return redirect(url_for(redirect_uri))
return function(**{**kwargs, "species": species})
return __with_species__
return __decorator__
def with_population(species_redirect_uri: str, redirect_uri: str):
"""Ensure the population actually exists."""
def __decorator__(function):
@wraps(function)
@with_species(redirect_uri=species_redirect_uri)
def __with_population__(**kwargs):
try:
species_id = int(kwargs["species_id"])
population_id = int(kwargs.get("population_id"))
select_population_uri = redirect(url_for(
redirect_uri, species_id=species_id))
if not bool(population_id):
flash("Expected population_id value to be present!",
"alert-danger")
return select_population_uri
with database_connection(app.config["SQL_URI"]) as conn:
population = population_by_species_and_id(
conn, species_id, population_id)
if not bool(population):
flash("Could not find population with that ID",
"alert-danger")
return select_population_uri
except ValueError as _verr:
logger.debug(
"Exception converting value to integer: %s",
kwargs.get("population_id"),
exc_info=True)
flash("Expected an integer for 'population_id' value.",
"alert-danger")
return select_population_uri
return function(**{**kwargs, "population": population})
return __with_population__
return __decorator__
def with_dataset(
species_redirect_uri: str,
population_redirect_uri: str,
redirect_uri: str,
dataset_by_id: Callable[
[Connection, int, int, int],
dict]
):
"""Ensure the dataset actually exists."""
def __decorator__(func):
@wraps(func)
@with_population(species_redirect_uri, population_redirect_uri)
def __with_dataset__(**kwargs):
try:
_spcid = int(kwargs["species_id"])
_popid = int(kwargs["population_id"])
_dsetid = int(kwargs.get("dataset_id"))
select_dataset_uri = redirect(url_for(
redirect_uri, species_id=_spcid, population_id=_popid))
if not bool(_dsetid):
flash("You need to select a valid 'dataset_id' value.",
"alert-danger")
return select_dataset_uri
with database_connection(app.config["SQL_URI"]) as conn:
dataset = dataset_by_id(conn, _spcid, _popid, _dsetid)
if not bool(dataset):
flash("You must select a valid dataset.",
"alert-danger")
return select_dataset_uri
except ValueError as _verr:
logger.debug(
"Exception converting 'dataset_id' to integer: %s",
kwargs.get("dataset_id"),
exc_info=True)
flash("Expected 'dataset_id' value to be an integer."
"alert-danger")
return select_dataset_uri
return func(**{**kwargs, "dataset": dataset})
return __with_dataset__
return __decorator__
|
