Rework to provide directory instead of file

This commit is contained in:
Aloïs Micard 2024-11-14 17:34:56 +01:00
parent 110c010408
commit 414e269587
4 changed files with 16 additions and 5 deletions

1
.gitignore vendored Normal file
View file

@ -0,0 +1 @@
*.pyc

View file

@ -2,6 +2,8 @@ services:
mongo: mongo:
image: mongo:latest image: mongo:latest
restart: always restart: always
ports:
- "27017:27017"
environment: environment:
MONGO_INITDB_ROOT_USERNAME: root MONGO_INITDB_ROOT_USERNAME: root
MONGO_INITDB_ROOT_PASSWORD: example MONGO_INITDB_ROOT_PASSWORD: example

View file

@ -2,6 +2,7 @@ import hashlib
import os import os
import subprocess import subprocess
import sys import sys
from pathlib import Path
from pymongo import MongoClient from pymongo import MongoClient
@ -28,11 +29,18 @@ if __name__ == '__main__':
database = mongo.exif_database database = mongo.exif_database
collection = database.pictures collection = database.pictures
picture_metadata = _execute_exiftool(sys.argv[1]) pictures_metadata = []
# Append MongoDB identifier for file in Path(sys.argv[1]).rglob("*.ARW"):
picture_metadata['_id'] = hashlib.sha256(sys.argv[1].encode('utf-8')).hexdigest() filename = os.fsdecode(file)
picture_metadata['path'] = sys.argv[1] print(filename)
picture_metadata = _execute_exiftool(filename)
pictures_metadata.append(picture_metadata)
# Append MongoDB identifier
picture_metadata['_id'] = hashlib.sha1(filename.lower().encode('utf-8')).hexdigest()
picture_metadata['path'] = sys.argv[1]
# Insert into MongoDB # Insert into MongoDB
collection.insert_one(picture_metadata) collection.insert_many(pictures_metadata)