Configurate streams amount from bridge config

Author: oleksiyVeretiuk

openprocurement/auction/bridge_utils/managers.py

@@ -82,7 +82,6 @@ def get_streams(self, db, streams_id=STREAMS_ID):
         :return: amount of streams for auction per day
         :rtype: int
         """
-
         streams = db.get(streams_id, deepcopy(DEFAULT_STREAMS_DOC))
         return streams.get(self.streams_key, DEFAULT_STREAMS_DOC[self.streams_key])
 

openprocurement/auction/databridge.py

@@ -23,7 +23,7 @@
     IAuctionDatabridge, IAuctionsManager
 from openprocurement.auction.bridge_utils.design import sync_design
 from openprocurement.auction.bridge_utils.managers import MANAGERS_MAPPING
-from openprocurement.auction.bridge_utils.constants import WORKING_DAYS, CALENDAR_ID
+from openprocurement.auction.bridge_utils.constants import WORKING_DAYS, CALENDAR_ID, STREAMS_ID
 
 from openprocurement.auction.core import components
 from openprocurement.auction.utils import FeedItem, check_workers
@@ -85,6 +85,7 @@ def __init__(self, config, re_planning=False, debug=False):
 
         self.stream_db = Database(db_for_streams, session=Session(retry_delays=range(10)))
         self._set_holidays()
+        self._set_streams_limits()
         sync_design(self.stream_db)
 
         # Managers Mapping
@@ -102,6 +103,15 @@ def _set_holidays(self):
             del self.stream_db[CALENDAR_ID]
         self.stream_db.save(calendar)
 
+    def _set_streams_limits(self):
+        streams = self.config.get('main').get('streams', {})
+
+        stream_amount = {'_id': STREAMS_ID}
+        stream_amount.update(streams)
+        if STREAMS_ID in self.stream_db:
+            del self.stream_db[STREAMS_ID]
+        self.stream_db.save(stream_amount)
+
     def config_get(self, name):
         return self.config.get('main').get(name)
 

openprocurement/auction/tests/bridge_utils/auctions_data_bridge.yaml

@@ -7,6 +7,10 @@ main:
   auctions_db: database
   timezone: Europe/Kiev
   stream_db : test_stream_db
+  streams:
+    streams: 10
+    dutch_streams: 15
+    texas_streams: 20
   plugins:
     english:
       auction_worker_config: /home/oleksiy.veretiuk/QuintaGroup/projects/kadastral/openprocurement.auction.buildout/etc/auction_worker_defaults.yaml