Update sqlite3/Files/README.md

This commit is contained in:
2024-09-26 00:56:28 -04:00
parent aa17195c65
commit 653920d43a

View File

@@ -1,37 +1,40 @@
```python
import sqlite3, pandas as pd
class Files:
def __init__(self, db='files.sqlite'):
def __init__(self, db="files.sqlite"):
self.conn = sqlite3.connect(db, isolation_level=None)
self.conn.row_factory = sqlite3.Row
self('PRAGMA journal_mode=WAL')
self('CREATE TABLE IF NOT EXISTS files (filename TEXT PRIMARY KEY, content BLOB)')
self("PRAGMA journal_mode=WAL")
self("CREATE TABLE IF NOT EXISTS files (filename TEXT PRIMARY KEY, content BLOB)")
def __call__(self, *args) -> pd.DataFrame:
return pd.DataFrame(dict(row) for row in self.conn.execute(*args))
def __setitem__(self, filename, content) -> None:
assert isinstance(filename, str)
if isinstance(content, str): content = content.encode()
if isinstance(content, str):
content = content.encode()
assert isinstance(content, bytes)
self('INSERT OR REPLACE INTO files VALUES (?,?)', (filename, content))
self("INSERT OR REPLACE INTO files VALUES (?,?)", (filename, content))
def __getitem__(self, filename) -> bytes:
return self('SELECT content FROM files WHERE filename=?', (filename,)).content[0]
return self("SELECT content FROM files WHERE filename=?", (filename,)).content[0]
def __delitem__(self, filename) -> None:
self('DELETE FROM files WHERE filename=?', (filename,))
self("DELETE FROM files WHERE filename=?", (filename,))
def __iter__(self):
df = self('SELECT filename FROM files')
df = self("SELECT filename FROM files")
return iter([] if df.empty else df.filename)
def __contains__(self, filename: str) -> bool:
return not self('SELECT 1 FROM files WHERE filename=?', (filename,)).empty
return not self("SELECT 1 FROM files WHERE filename=?", (filename,)).empty
def __len__(self) -> int:
return self('SELECT COUNT(*) FROM files').iloc[0, 0]
return self("SELECT COUNT(*) FROM files").iloc[0, 0]
files = Files()
for filename in files: