Check if a file ID already exists before creating a new record

Author: Zsailer

jupyter_server_fileid/manager.py

@@ -311,19 +311,18 @@ def _from_normalized_path(self, path: Optional[str]) -> Optional[str]:
         return relpath
 
     def _create(self, path: str) -> str:
-        path = self._normalize_path(path)
-        id = self._uuid()
-        self.con.execute("INSERT INTO Files (id, path) VALUES (?, ?)", (id, path))
-        return id
-
-    def index(self, path: str) -> str:
         path = self._normalize_path(path)
         row = self.con.execute("SELECT id FROM Files WHERE path = ?", (path,)).fetchone()
         existing_id = row and row[0]
 
         if existing_id:
             return existing_id
+        
+        id = self._uuid()
+        self.con.execute("INSERT INTO Files (id, path) VALUES (?, ?)", (id, path))
+        return id
 
+    def index(self, path: str) -> str:
         # create new record
         id = self._create(path)
         self.con.commit()
@@ -613,6 +612,7 @@ def _sync_file(self, path, stat_info):
             "SELECT id, path, crtime FROM Files WHERE ino = ?", (stat_info.ino,)
         ).fetchone()
 
+
         # if ino is not in database, return None
         if src is None:
             return None
@@ -626,6 +626,7 @@ def _sync_file(self, path, stat_info):
         # otherwise update existing record with new path, moving any indexed
         # children if necessary. then return its id
         self._update(id, path=path)
+
         if stat_info.is_dir and old_path != path:
             self._move_recursive(old_path, path)
             self._update_cursor = True
@@ -672,6 +673,17 @@ def _create(self, path, stat_info):
         dangerous and may throw a runtime error if the file is not guaranteed to
         have a unique `ino`.
         """
+        # If the path exists 
+        existing_id, ino = None, None
+        row = self.con.execute("SELECT id, ino FROM Files WHERE path = ?", (path,)).fetchone()
+        if row: 
+            existing_id, ino = row
+
+        # If the file ID already exists and the current file matches our records
+        # return the file ID instead of creating a new one. 
+        if existing_id and stat_info.ino == ino:
+            return existing_id
+            
         id = self._uuid()
         self.con.execute(
             "INSERT INTO Files (id, path, ino, crtime, mtime, is_dir) VALUES (?, ?, ?, ?, ?, ?)",