def get()

in distilvit/gpt4.py [0:0]

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

    def get(self, ids):
        self._lock_table()
        try:
            placeholders = ", ".join(["%s"] * len(ids))
            query = f"SELECT json_key, payload FROM {self.table} WHERE json_key IN ({placeholders})"
            self.cursor.execute(query, ids)
            result = self.cursor.fetchall()
            return {row[0]: json.loads(row[1]) for row in result}
        finally:
            self._unlock_table()