commit d79ab06b0e6dd39616dfe84b6b16411f24719f78 · zenfyr.dev/xpost

+12 -4
main.py
···

       107
       107
        
               outputs.append(OUTPUTS[output_settings['type']](input, output_settings, db_worker))

     

       108
       108
        
           

     

       109
       109
        
           LOGGER.info('Starting task worker...')

     

       110
       110
       -
           task_queue = queue.Queue()

     

       111
       111
       -
           def worker():

     

       110
       110
       +
           def worker(queue: queue.Queue):

     

       112
       111
        
               while True:

     

       113
       113
       -
                   task = task_queue.get()

     

       112
       112
       +
                   task = queue.get()

     

       114
       113
        
                   if task is None:

     

       115
       114
        
                       break

     

       115
       115
       +
                   

     

       116
       116
        
                   try:

     

       117
       117
        
                       task()

     

       118
       118
       +
                       queue.task_done()

     

       118
       119
        
                   except Exception as e:

     

       119
       120
        
                       LOGGER.error(f"Exception in worker thread!\n{e}")

     

       120
       121
        
                       traceback.print_exc()

     

       121
       121
       -
           thread = threading.Thread(target=worker, daemon=True)

     

       122
       122
       +
           

     

       123
       123
       +
           task_queue = queue.Queue()

     

       124
       124
       +
           thread = threading.Thread(target=worker, args=(task_queue,))

     

       122
       125
        
           thread.start()

     

       126
       126
       +
           

     

       123
       127
        
           LOGGER.info('Connecting to %s...', input.service)

     

       124
       128
        
           asyncio.run(input.listen(outputs, lambda x: task_queue.put(x)))

     

       129
       129
       +
           

     

       130
       130
       +
           task_queue.join()

     

       131
       131
       +
           task_queue.put(None)

     

       132
       132
       +
           thread.join()

     

       125
       133
        
           

     

       126
       134
        
       

     

       127
       135
        
       if __name__ == "__main__":