@@ -163,14 +163,14 @@ def filterByDate(queryset, request=None, datemin=None, datemax=None):
163163 # datemax = request.data.get("datemax", datemax)
164164
165165 if datemin :
166- queryset = django_filters .DateFilter (field_name = "dateTime" , lookup_expr = "gte" ). filter (
167- queryset , datemin
168- )
166+ queryset = django_filters .DateFilter (
167+ field_name = "dateTime" , lookup_expr = "gte"
168+ ). filter ( queryset , datemin )
169169
170170 if datemax :
171- queryset = django_filters .DateFilter (field_name = "dateTime" , lookup_expr = "lt" ). filter (
172- queryset , datemax
173- )
171+ queryset = django_filters .DateFilter (
172+ field_name = "dateTime" , lookup_expr = "lt"
173+ ). filter ( queryset , datemax )
174174
175175 return (queryset , datemin , datemax )
176176
@@ -330,30 +330,37 @@ def by_root(request, format=None):
330330 }
331331 )
332332
333+
333334@api_view (("POST" ,))
334335def query (request , format = None ):
335336 sql_err , sql = get_parameter (request , "sql" )
336337 token_err , token = get_parameter (request , "token" )
337338 if sql_err or token_err :
338- return response .Response (status = 400 , data = f"Invalid Parameters: { [x for x in [sql_err , token_err ] if x ]} " )
339+ return response .Response (
340+ status = 400 ,
341+ data = f"Invalid Parameters: { [x for x in [sql_err , token_err ] if x ]} " ,
342+ )
339343 verified = verify_token (token )
340344 if not verified :
341345 return response .Response (status = 401 , data = "UNAUTHORIZED" )
342- conn = connections ["readonly" ]
346+ conn = connections ["readonly" ]
343347 with conn .cursor () as cur :
344348 cur .execute (sql )
345349 res = cur .fetchall ()
346350 return response .Response (res )
347351
352+
348353def get_parameter (request , param ):
349354 val = request .POST .get (param )
350355 err = ""
351356 if not val :
352357 err = f"No { param } parameter provided"
353358 return err , val
354359
360+
355361def verify_token (token ):
356- return token == environ ["QUERY_SECRET_KEY" ]
362+ return token == environ ["QUERY_SECRET_KEY" ]
363+
357364
358365class FeatureViewSet (viewsets .ModelViewSet ):
359366 """
0 commit comments