a fD @sRddlZddlZddlmZddlmZddlmZddlmZGdddZ dS)N)datetime)randint) Environment) pubsub_v1c@s(eZdZdZddZddZddZdS) CloudQueueServicezService to handle interaction with a Google Cloud PubSub queue. This service manages the publishing of tasks to a Google Cloud PubSub queue. Attributes: publisher: The PubSub client used to interact with Google Cloud PubSub. cCs6tjs(tjtjtddtjtjd<t |_ dS)z]Initializes the CloudQueueService with appropriate environment variables and a PubSub client.z..GOOGLE_APPLICATION_CREDENTIALSN) r LIVE_SERVERospathjoindirname__file__gcloud_creds_filenameenvironrZPublisherClient publisherselfrO/var/www/html/python-backend/backend/services/scheduling/cloud_queue_service.py__init__s"zCloudQueueService.__init__cCs"ttdttddS)zGenerates a unique task id based on current time and a random number. Returns: str: The unique task id as a string. iri)strrutcnow timestamprrrrrgenerate_task_idsz"CloudQueueService.generate_task_idcCsf|jtj|}||d}t|}|d}ddi}|jj||fi|}t d| dS)aSchedules a task by publishing it to a specified PubSub topic. The function creates a task with a unique id and the provided description, then publishes it to the provided topic id. It adds a task type attribute ("compute") to the message. Args: topic_id (str): The id of the PubSub topic to which the task should be published. task_description (dict): The description of the task to be scheduled. Returns: None )task_idtask_descriptionzutf-8Z task_typeZcomputezPublished message ID N) r topic_pathrGCLOUD_PROJECT_IDrjsondumpsencodeZpublishprintresult)rZtopic_idrrtaskdata attributesfuturerrr schedule_task s   zCloudQueueService.schedule_taskN)__name__ __module__ __qualname____doc__rrr'rrrrr sr) rr rrandomr environmentrZ google.cloudrrrrrrs