CLI: Add cli interface.

Author: nir-mo

cli.py

@@ -0,0 +1,52 @@
+import logging
+import os
+
+import click
+
+from ioccc_winners_db import IOCCCWinnersDB, build_sqllite_db
+
+DEFAULT_IOCCC_WINNERS_DIRECTORY = os.path.join(os.path.dirname(__file__), "winner")
+DEFAULT_IOCCC_SQLITE_DB_PATH = os.path.join(os.path.dirname(__file__), "ioccc_winners.sqlite")
+
+
+@click.command()
+@click.option(
+    '--ioccc_winners_directory',
+    default=DEFAULT_IOCCC_WINNERS_DIRECTORY,
+    help='IOCCC winners directory, should be a clone of the following github: https://github.com/ioccc-src/winner.git.'
+)
+@click.option(
+    '--output_file',
+    default=DEFAULT_IOCCC_SQLITE_DB_PATH,
+    help='Path to the output db.'
+)
+@click.option(
+    '--force',
+    is_flag=True,
+    default=False,
+    help='Overrides output_file if its already exists.'
+)
+@click.option(
+    '--verbose',
+    is_flag=True,
+    default=False,
+    help='Verbose mode.'
+)
+def build_db(ioccc_winners_directory, output_file, force, verbose):
+    logging.basicConfig(level=logging.DEBUG if verbose else logging.INFO)
+    if os.path.exists(output_file):
+        if force:
+            logging.debug(f"File {output_file} already exists! removing it...")
+            os.remove(output_file)
+        else:
+            logging.error(f"File {output_file} already exists! Use --force to override it.")
+            exit(1)
+
+    logging.info(f"Building DB: {output_file}")
+    winners_db = IOCCCWinnersDB(ioccc_winners_directory)
+    build_sqllite_db(winners_db.get_all_entries(), output_file)
+    logging.info("Done!")
+
+
+if __name__ == '__main__':
+    build_db()

ioccc_winners_db.py

@@ -138,7 +138,6 @@ class _EntryDetails:
 
 
 def build_sqllite_db(entries: Iterator[IOCCCWinnerEntry], output_file: str):
-    # Connect to the SQLite database (or create it if it doesn't exist)
     with sqlite3.connect(output_file) as conn:
         # Create the "winners" table
         conn.execute("""
@@ -164,11 +163,3 @@ def build_sqllite_db(entries: Iterator[IOCCCWinnerEntry], output_file: str):
                 """,
                 (entry.name, entry.year, entry.spoiler, entry.prog(), hint)
             )
-
-
-if __name__ == "__main__":
-    logging.basicConfig(level=logging.DEBUG)
-    _logger.info("Building DB...")
-    winners_db = IOCCCWinnersDB()
-    build_sqllite_db(winners_db.get_all_entries(), "ioccc_winners.sqlite")
-    _logger.info("Done!")

requirements.txt

@@ -0,0 +1 @@
+click==8.1.4