File name
Commit message
Commit date
2024-05-28
2024-05-20
from flask import Flask, request
from flask_restx import Api, Resource, fields
import os
from datetime import datetime
import base64
# from yoloseg.inference_ import Inference
app = Flask(__name__)
api = Api(app, version='1.0', title='CCTV Image Upload API',
description='A simple API for receiving CCTV images')
# Namespace definition
ns = api.namespace('cctv', description='CCTV operations')
# Define the expected model for incoming data
image_upload_model = api.model('ImageUpload', {
'image': fields.String(required=True, description='Image file', dt='File'),
'x-cctv-info': fields.String(required=False, description='CCTV identifier'),
'x-time-sent': fields.String(required=False, description='Time image was sent'),
'x-cctv-latitude': fields.String(required=False, description='Latitude of CCTV'),
'x-cctv-longitude': fields.String(required=False, description='Longitude of CCTV')
})
# Define the directory where images will be saved
IMAGE_DIR = "network_test"
if not os.path.exists(IMAGE_DIR):
os.makedirs(IMAGE_DIR)
@ns.route('/infer', )
class ImageUpload(Resource):
# @ns.expect(image_upload_model, validate=True)
@ns.response(200, 'Success')
@ns.response(400, 'Validation Error')
def post(self):
if 'file' not in request.files:
ns.abort(400, 'No image part in the request')
image = request.files['file']
cctv_info = base64.b64decode(request.headers.get('x-cctv-name', '')).decode('UTF-8')
print(cctv_info)
time_sent = request.headers.get('x-time-sent', '')
cctv_latitude = request.headers.get('x-cctv-latitude', 'Not provided')
cctv_longitude = request.headers.get('x-cctv-longitude', 'Not provided')
if image.filename == '':
ns.abort(400, 'No selected image')
# Use current timestamp to avoid filename conflicts
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
filename = f"{timestamp}_{cctv_info}.png"
image_path = os.path.join(IMAGE_DIR, filename)
image.save(image_path)
print("successful")
return {"message": f"Image {filename} uploaded successfully!"}
if __name__ == '__main__':
app.run(debug=True, port=12345)