Coverage for mindsdb / interfaces / query_context / query_task.py: 0%

1from mindsdb.api.executor.sql_query import SQLQuery 

2from mindsdb.interfaces.query_context.context_controller import query_context_controller 

3from mindsdb.api.executor.controllers.session_controller import SessionController 

4from mindsdb.interfaces.tasks.task import BaseTask 

5 

6 

7class QueryTask(BaseTask): 

8 def __init__(self, *args, **kwargs): 

9 super().__init__(*args, **kwargs) 

10 self.query_id = self.object_id 

11 

12 def run(self, stop_event): 

13 

14 try: 

15 session = SessionController() 

16 SQLQuery(None, query_id=self.query_id, session=session, stop_event=stop_event) 

17 finally: 

18 # clear task 

19 query_context_controller.get_query(self.query_id).remove_from_task()