View GitHub Repository distributer

Schedule Inspection
Subscribe

QueueManager.__init__()   A
last analyzed

↳ Parent: QueueManager

Complexity

Conditions 2

Size

Total Lines 19

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 13
CRAP Score 2

Importance

Changes 1
Bugs 0 Features 0
Metric Value
c 1
b 0
f 0
dl 0
loc 19
ccs 13
cts 13
cp 1
rs 9.4285
cc 2
crap 2
1 1 
import os
2 

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    3
                

                                    
                             1                         
                
                
                
from impulsare_config import Reader as ConfigReader

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    4
                

                                    
                             1                         
                
                
                
from impulsare_logger import Logger

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    5
                

                                    
                             1                         
                
                
                
from redis import Redis

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    6
                

                                    
                             1                         
                
                
                
from rq import Queue as RedisQueue

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    7
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    8
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                                                                
            
                                    
            
            

                
                    9
                

                                    
                             1                         
                
                
                
class QueueManager():

            


            
                                                                        
                            
            
                                    
            
            

                
                    10
                

                                    
                             1                         
                
                
                
    def __init__(self, config_file: str, listener: str):

            


            
                                                                        
                            
            
                                    
            
            

                
                    11
                

                                    
                                                    
                
                
                
        """Init the Queue from config parameters"""

            


            
                                                                        
                            
            
                                    
            
            

                
                    12
                

                                    
                                                    
                
                
                

            


            
                                                                        
                            
            
                                    
            
            

                
                    13
                

                                    
                             1                         
                
                
                
        base_path = os.path.abspath(os.path.dirname(__file__))

            


            
                                                                        
                            
            
                                    
            
            

                
                    14
                

                                    
                             1                         
                
                
                
        config_specs = base_path + '/static/specs.yml'

            


            
                                                                        
                            
            
                                    
            
            

                
                    15
                

                                    
                             1                         
                
                
                
        config_default = base_path + '/static/default.yml'

            


            
                                                                        
                            
            
                                    
            
            

                
                    16
                

                                    
                                                    
                
                
                

            


            
                                                                        
                            
            
                                    
            
            

                
                    17
                

                                    
                             1                         
                
                
                
        config = ConfigReader().parse(config_file, config_specs, config_default)

            


            
                                                                        
                            
            
                                    
            
            

                
                    18
                

                                    
                             1                         
                
                
                
        config_listener = config.get('distributer')

            


            
                                                                        
                            
            
                                    
            
            

                
                    19
                

                                    
                                                    
                
                
                

            


            
                                                                        
                            
            
                                    
            
            

                
                    20
                

                                    
                             1                         
                
                
                
        redis = Redis(config_listener['host'], config_listener['port'])

            


            
                                                                        
                            
            
                                    
            
            

                
                    21
                

                                    
                                                    
                
                
                

            


            
                                                                        
                            
            
                                    
            
            

                
                    22
                

                                    
                             1                         
                
                
                
        if listener not in config:

            


            
                                                                        
                            
            
                                    
            
            

                
                    23
                

                                    
                             1                         
                
                
                
            raise KeyError('You must have a key {} in your config with a sub-key queue'.format(listener))

            


            
                                                                        
                            
            
                                    
            
            

                
                    24
                

                                    
                                                    
                
                
                

            


            
                                                                        
                            
            
                                    
            
            

                
                    25
                

                                    
                             1                         
                
                
                
        self._logger = Logger('distributer', config_file)

            


            
                                                                        
                            
            
                                    
            
            

                
                    26
                

                                    
                             1                         
                
                
                
        self._logger.log.debug('Distributer QueueManager called')

            


            
                                                                        
                            
            
                                    
            
            

                
                    27
                

                                    
                             1                         
                
                
                
        self._logger.log.debug('Connect to queue {}'.format(config[listener]['queue']))

            


            
                                                                        
                            
            
                                    
            
            

                
                    28
                

                                    
                             1                         
                
                
                
        self._queue = RedisQueue(config[listener]['queue'], connection=redis)

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    29
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    30
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    31
                

                                    
                             1                         
                
                
                
    def add(self, method: str, item: str, job: str):

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    32
                

                                    
                                                    
                
                
                
        """Put item into the queue."""

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    33
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                                                                
            
                                    
            
            

                
                    34
                

                                    
                                                    
                
                
                
        return self._queue.enqueue(method, item, job=job)

            


            
                                                        
            
                                    
            
            

                
                    35