]> git.treefish.org Git - photosort.git/blob - src/locator.py
fixing db creation
[photosort.git] / src / locator.py
1 import logging
2 import os
3 import sqlite3
4 import time
5
6 import misc
7
8 class Locator:
9     def __init__(self, base_dir, exclude_dir, db_file):
10         self._base_dir = base_dir
11         self._exclude_dir = exclude_dir
12         if not os.path.isfile(db_file):
13             self._conn = sqlite3.connect(db_file)
14             self._create_db()
15         else:
16             self._conn = sqlite3.connect(db_file)
17         logging.info("Opened locator for %s.", self._base_dir)
18
19     def locate_file(self, name, size, meta_time):
20         c = self._conn.cursor()
21         c.execute("SELECT file_dir FROM cache WHERE name=? AND size=? AND meta_time=?",
22                   (name, size, meta_time))
23         cached = c.fetchone()
24         if cached:
25             if os.path.isfile(cached[0]) and \
26                os.path.getsize(cached[0]) == size and \
27                misc.extract_meta_time(cached[0]) == meta_time:
28                 self._update_cache_access_time(name, size, meta_time)
29                 return cached[0]
30             else:
31                 self._remove_from_cache(name, size, meta_time)
32         else:
33             file_dir = misc.find_file(name, size, meta_time)
34             if file_dir:
35                 self._clean_cache()
36                 self._add_to_cache(name, size, meta_time, file_dir)
37             return file_dir
38
39     def close(self):
40         logging.info("Closing locator for %s...", self._base_dir)
41         self._conn.close()
42
43     def _update_cache_access_time(self, name, size, meta_time):
44         c = self._conn.cursor()
45         c.execute("UPDATE cache SET access_time=? WHERE name=? AND size=? AND meta_time=?",
46                   (int(time.time()), name, size, meta_time))
47         self._conn.commit()
48
49     def _remove_from_cache(self, name, size, meta_time):
50         c = self._conn.cursor()
51         c.execute("DELETE FROM cache WHERE name=? AND size=? AND meta_time=?",
52                   (name, size, meta_time))
53         self._conn.commit()
54
55     def _add_to_cache(self, name, size, meta_time, file_dir):
56         c = self._conn.cursor()
57         c.execute('''INSERT INTO cache (name, size, meta_time, file_dir, access_time)
58                      VALUES (?, ?, ?, ?, ?)''',
59                   (name, size, meta_time, file_dir, int(time.time())))
60         self._conn.commit()
61
62     def _clean_cache(self):
63         c = self._conn.cursor()
64         c.execute("DELETE FROM cache WHERE access_time<?", (int(time.time()) - 604800))
65         self._conn.commit()
66
67     def _create_db(self):
68         logging.info("Creating locator database for %s...", self._base_dir)
69         c = self._conn.cursor()
70         c.execute('''CREATE TABLE cache
71                        (name TEXT, size INTEGER, meta_time INTEGER, file_dir TEXT,
72                         access_time INTEGER),
73                      PRIMARY KEY (name, size, meta_time)''')
74         self._conn.commit()