simple documentation comments

This commit is contained in:
aradhyacp
2025-12-30 11:59:02 +05:30
parent 8e72911f54
commit 4a380d277e
+14 -51
View File
@@ -112,25 +112,33 @@ def upload_image() -> tuple[Response, int]:
submission_hash = hashlib.md5(submission_data).hexdigest() submission_hash = hashlib.md5(submission_data).hexdigest()
submission_path = RESULT_FOLDER / img_hash / submission_hash submission_path = RESULT_FOLDER / img_hash / submission_hash
# Check if hash is present on the DB
submission: Submission = Submission.query.filter_by( submission: Submission = Submission.query.filter_by(
hash=submission_hash hash=submission_hash
).first() ).first()
# Only return if BOTH the file exists AND the DB entry exists
if submission_path.exists() and submission is not None: if submission_path.exists() and submission is not None:
return jsonify({"submission_hash": submission.hash}),200 return jsonify({"submission_hash": submission.hash}),200
# Self-Healing Logic
new_img_path = RESULT_FOLDER / img_hash / image_name new_img_path = RESULT_FOLDER / img_hash / image_name
# Try to find image in DB by hash
sub_img = Image.query.filter_by( sub_img = Image.query.filter_by(
hash=img_hash hash=img_hash
).first() ).first()
# If DB entry is missing (even if file exists on disk), create it!
if sub_img is None: if sub_img is None:
if not new_img_path.parent.exists() or not new_img_path.exists(): if not new_img_path.parent.exists() or not new_img_path.exists():
new_img_path.parent.mkdir(parents=True, exist_ok=True) new_img_path.parent.mkdir(parents=True, exist_ok=True)
with open(new_img_path,"wb") as f: with open(new_img_path,"wb") as f:
f.write(image_data) f.write(image_data)
# Recreate the missing Database Entry
sub_img = Image( sub_img = Image(
file=str(new_img_path), file=str(new_img_path),
hash=img_hash, hash=img_hash,
@@ -142,12 +150,15 @@ def upload_image() -> tuple[Response, int]:
db.session.add(sub_img) db.session.add(sub_img)
db.session.commit() db.session.commit()
# Recreate the missing Database Entry
sub_img.upload_count += 1 sub_img.upload_count += 1
db.session.commit() db.session.commit()
# Create new Submission entry
submission_path.mkdir(parents=True, exist_ok=True) submission_path.mkdir(parents=True, exist_ok=True)
# Only create a new submission if it does not already exist
# New submission for new image
if not submission: if not submission:
submission = Submission( submission = Submission(
filename=image.filename, filename=image.filename,
@@ -161,62 +172,14 @@ def upload_image() -> tuple[Response, int]:
db.session.add(submission) db.session.add(submission)
db.session.commit() db.session.commit()
# Start the analysis jobs
queue.enqueue("aperisolve.workers.analyze_image", submission.hash, job_timeout=300) queue.enqueue("aperisolve.workers.analyze_image", submission.hash, job_timeout=300)
# Return submission identifier
return jsonify({"submission_hash": submission.hash}), 200 return jsonify({"submission_hash": submission.hash}), 200
# # If submission already exists, return submission hash
# if submission_path.exists():
# submission: Submission = Submission.query.filter_by(
# hash=submission_hash
# ).first()
# return jsonify({"submission_hash": submission.hash}), 200
# # If image is new, create a new Image entry
# new_img_path = RESULT_FOLDER / img_hash / image_name
# if not new_img_path.parent.exists():
# new_img_path.parent.mkdir(parents=True, exist_ok=True)
# with open(new_img_path, "wb") as f: # Write file to disk
# f.write(image_data)
# new_img = Image(
# file=str(new_img_path),
# hash=img_hash,
# size=len(image_data),
# upload_count=0,
# first_submission_date=datetime.now(timezone.utc),
# last_submission_date=datetime.now(timezone.utc),
# )
# db.session.add(new_img) # pylint: disable=no-member
# db.session.commit() # pylint: disable=no-member
# sub_img = Image.query.filter_by(hash=img_hash).first() # type: ignore
# sub_img.upload_count += 1
# db.session.commit() # pylint: disable=no-member
# # Create new Submission entry
# submission_path.mkdir(parents=True, exist_ok=True)
# submission = Submission(
# filename=image.filename,
# password=password,
# deep_analysis=deep_analysis,
# hash=submission_hash,
# status="pending",
# date=time.time(),
# image_hash=sub_img.hash, # type: ignore
# )
# db.session.add(submission) # pylint: disable=no-member
# db.session.commit() # pylint: disable=no-member
# # Start the analysis jobs
# queue.enqueue("aperisolve.workers.analyze_image", submission.hash, job_timeout=300)
# return jsonify({"submission_hash": submission.hash}), 200
@app.route("/status/<hash_val>", methods=["GET"]) @app.route("/status/<hash_val>", methods=["GET"])
def get_status(hash_val: str) -> Response: def get_status(hash_val: str) -> Response:
"""Get the processing status of a submission.""" """Get the processing status of a submission."""