def set()

in distilvit/gpt4.py [0:0]

• 16 lines of code
• 3 McCabe index (conditional complexity)

    def set(self, ids, payloads):
        if len(ids) != len(payloads):
            print(ids)
            print(payloads)
            raise ValueError("Length of ids and payloads must match")

        self._lock_table()
        try:
            for i in range(len(ids)):
                query = (
                    f"INSERT INTO {self.table} (json_key, payload) VALUES (%s, %s) "
                    f"ON DUPLICATE KEY UPDATE payload = VALUES(payload)"
                )
                self.cursor.execute(query, (ids[i], json.dumps(payloads[i])))
            self.conn.commit()
        finally:
            self._unlock_table()