From 1bbb918c1d5a4009e2d0b302dff689fb7a41ed97 Mon Sep 17 00:00:00 2001 From: jay817 Date: Thu, 26 Sep 2024 01:10:06 -0400 Subject: [PATCH] Update sqlite3/Files/README.md --- sqlite3/Files/README.md | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) 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])