CoverageSqliteData._file_id: fall back to SELECT

Fixes #702
nedbat · Oct 23, 2018 · 1fec608 · 1fec608
1 parent 6c14ffb
commit 1fec608
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/coverage/sqldata.py b/coverage/sqldata.py
@@ -182,8 +182,16 @@ def _file_id(self, filename, add=False):
         if filename not in self._file_map:
             if add:
                 with self._connect() as con:
-                    cur = con.execute("insert into file (path) values (?)", (filename,))
-                    self._file_map[filename] = cur.lastrowid
+                    try:
+                        cur = con.execute("insert into file (path) values (?)", (filename,))
+                    except CoverageException as exc:
+                        if "UNIQUE constraint failed: file.path" in str(exc):
+                            cur = self._db.execute("select id from file where path = (?)", (filename,))
+                            self._file_map[filename] = cur.fetchone()[0]
+                        else:
+                            raise
+                    else:
+                        self._file_map[filename] = cur.lastrowid
         return self._file_map.get(filename)
 
     def _context_id(self, context):