diff --git a/sqlite3/Files/README.md b/sqlite3/Files/README.md index 8e2df63..e65dfff 100644 --- a/sqlite3/Files/README.md +++ b/sqlite3/Files/README.md @@ -1,10 +1,11 @@ ```python -import sqlite3, pandas as pd - +import pandas as pd class Files: - def __init__(self, db="files.sqlite"): + def __init__(self, db='files.sqlite'): + import sqlite3 + if not db.endswith('.sqlite'): db += '.sqlite' self.conn = sqlite3.connect(db, isolation_level=None) self.conn.row_factory = sqlite3.Row self("PRAGMA journal_mode=WAL") @@ -36,6 +37,7 @@ class Files: def __len__(self) -> int: return self("SELECT COUNT(*) FROM files").iloc[0, 0] + files = Files() for filename in files: print(files[filename])