61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import os
|
||
|
|
import re
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from avdb.db import DBOps, DBConnection
|
||
|
|
|
||
|
|
def process_single_file(db_ops, file_path):
|
||
|
|
"""
|
||
|
|
Process a single image file.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
db_ops (DBOps): An instance of DBOps for database operations.
|
||
|
|
file_path (str): Path to the image file.
|
||
|
|
"""
|
||
|
|
print(f"Processing file: {file_path}")
|
||
|
|
db_ops.process_image_analysis(file_path)
|
||
|
|
|
||
|
|
def process_directory(db_ops, directory_path):
|
||
|
|
"""
|
||
|
|
Process all image files in a directory.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
db_ops (DBOps): An instance of DBOps for database operations.
|
||
|
|
directory_path (str): Path to the directory containing image files.
|
||
|
|
"""
|
||
|
|
print(f"Processing directory: {directory_path}")
|
||
|
|
for root, _, files in os.walk(directory_path):
|
||
|
|
# print(files)
|
||
|
|
# print(root)
|
||
|
|
for file in files:
|
||
|
|
m = re.match(r'img_\d{4}\.jpeg', file.lower())
|
||
|
|
if m is not None:
|
||
|
|
file_path = os.path.join(root, file)
|
||
|
|
process_single_file(db_ops, file_path)
|
||
|
|
|
||
|
|
def main():
|
||
|
|
# Parse command-line arguments
|
||
|
|
parser = argparse.ArgumentParser(description="Run image analysis for a single file or all files in a directory.")
|
||
|
|
parser.add_argument("path", help="Path to a single image file or a directory containing image files.")
|
||
|
|
parser.add_argument("--dbname", default="testdb", help="Name of the database to use (default: testdb).")
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
# Initialize database connection and operations
|
||
|
|
with DBConnection(args.dbname).get_connection() as db_conn:
|
||
|
|
db_ops = DBOps(db_conn)
|
||
|
|
|
||
|
|
# Determine if the path is a file or directory
|
||
|
|
if os.path.isfile(args.path):
|
||
|
|
process_single_file(db_ops, args.path)
|
||
|
|
elif os.path.isdir(args.path):
|
||
|
|
process_directory(db_ops, args.path)
|
||
|
|
else:
|
||
|
|
print(f"Error: Path '{args.path}' does not exist.")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|