def conn()

in distilvit/gpt4.py [0:0]

• 15 lines of code
• 2 McCabe index (conditional complexity)

    def conn(self):
        if self._conn is not None:
            return self._conn

        self._conn = mysql.connector.connect(
            host=self.host,
            user=self.user,
            password=self.password,
            database=self.database,
        )
        self._cursor = self._conn.cursor()
        self._cursor.execute(
            f"""
            CREATE TABLE IF NOT EXISTS {self.table} (
                id INT AUTO_INCREMENT PRIMARY KEY,
                json_key VARCHAR(255) NOT NULL UNIQUE,
                payload JSON NOT NULL
            )
        """
        )
        self._conn.commit()
        return self._conn