fix injector

wuan · Oct 26, 2018 · 85b4bb4 · 85b4bb4
1 parent 09a18a3
commit 85b4bb4
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/blitzortung/service/histogram.py b/blitzortung/service/histogram.py
@@ -24,8 +24,8 @@
 
 
 class HistogramQuery(object):
-    @inject(strike_query_builder=blitzortung.db.query_builder.Strike)
-    def __init__(self, strike_query_builder):
+    @inject
+    def __init__(self, strike_query_builder: blitzortung.db.query_builder.Strike):
         self.strike_query_builder = strike_query_builder
 
     def create(self, connection, minute_length, minute_offset, region=None, envelope=None, count_threshold=0):

diff --git a/blitzortung/service/strike.py b/blitzortung/service/strike.py
@@ -39,8 +39,8 @@ def __init__(self, statsd_client, end_time):
 
 
 class StrikeQuery(object):
-    @inject(strike_query_builder=db.query_builder.Strike, strike_mapper=db.mapper.Strike)
-    def __init__(self, strike_query_builder, strike_mapper):
+    @inject
+    def __init__(self, strike_query_builder: db.query_builder.Strike, strike_mapper: db.mapper.Strike):
         self.strike_query_builder = strike_query_builder
         self.strike_mapper = strike_mapper
 

diff --git a/blitzortung/service/strike_grid.py b/blitzortung/service/strike_grid.py
@@ -38,8 +38,8 @@ def __init__(self, statsd_client, grid_parameters, time_interval):
 
 
 class StrikeGridQuery(object):
-    @inject(strike_query_builder=db.query_builder.Strike)
-    def __init__(self, strike_query_builder):
+    @inject
+    def __init__(self, strike_query_builder: db.query_builder.Strike):
         self.strike_query_builder = strike_query_builder
 
     def create(self, grid_parameters, minute_length, minute_offset, count_threshold, connection, statsd_client):