item_manager.py

from retrying import retry
from gql import gql
from logger import logger
from queries import get_items_query, search_items_query, get_item_by_id_query, check_duplicate_query, create_comment_query

class ItemManager:
    def __init__(self, client):
        self.client = client

    @retry(wait_exponential_multiplier=3000, wait_exponential_max=10000, stop_max_attempt_number=3, retry_on_exception=lambda e: True)
    def execute(self, query, variables=None, attempt=1):
        gql_query = gql(query)
        try:
            result = self.client.execute(gql_query, variable_values=variables)
            if attempt > 1:
                logger.info(f"Query succeeded on attempt {attempt}.")
            return result
        except Exception as e:
            logger.error(f"Error executing query on attempt {attempt}: {str(e)}")
            if attempt < 3:  # If it's not the last attempt
                return self.execute(query, variables, attempt + 1)
            raise


    def create_comment(self, parent_id, text):
        variables = {
            "text": text,
            "parentId": parent_id
        }
        response = self.execute(create_comment_query, variables)
        return response["upsertComment"]["id"]

    def get_items(self, limit=10, cursor=None, sort="NEW", type=None, sub=None, name=None, when=None, by=None):
        variables = {
            "limit": limit,
            "cursor": cursor,
            "sort": sort,
            "type": type,
            "sub": sub,
            "name": name,
            "when": when,
            "by": by
        }
        response = self.execute(get_items_query, variables)
        return {
            "items": response["items"],
            "cursor": cursor,
            "limit": limit
        }

    def search_items(self, q, sub=None, cursor=None, what=None, sort=None, when=None, limit=10):
        variables = {
            "q": q,
            "sub": sub,
            "cursor": cursor,
            "what": what,
            "sort": sort,
            "when": when
        }
        response = self.execute(search_items_query, variables)
        return response['search']['items']

    def get_item_by_id(self, item_id):
        variables = {
            "id": item_id
        }
        return self.execute(get_item_by_id_query, variables)

    def check_duplicate(self, url):
        variables = {
            "url": url
        }
        return self.execute(check_duplicate_query, variables)