api.py

#api.py
from flask import Flask, request, jsonify
from flask_jwt_extended import JWTManager, create_access_token, jwt_required
from pymongo import MongoClient
from datetime import datetime
import os
from dotenv import load_dotenv

load_dotenv()
MONGO_URI = os.getenv("MONGO_URI")

app = Flask(__name__)
app.config["JWT_SECRET_KEY"] = os.getenv("JWT_SECRET_KEY", "supersecret")
jwt = JWTManager(app)

try:
    client = MongoClient(MONGO_URI)
    db = client["FaceSecureDB"]
    client.admin.command('ping')
    print("MongoDB bağlantısı başarılı!")
except Exception as e:
    print(f"MongoDB bağlantı hatası: {e}")
    db = None 

@app.route("/login", methods=["POST"])
def login():
    data = request.json
    username = data.get("username")
    password = data.get("password")
    if username == "admin" and password == "admin123":
        token = create_access_token(identity=username)
        return jsonify(access_token=token)
    return jsonify(msg="Hatalı kimlik bilgisi"), 401

@app.route("/users", methods=["POST"])
@jwt_required()
def add_user():
    if db is None:
        return jsonify(msg="Veritabanı bağlantısı kurulamadı."), 500

    user = request.json.get("user")
    if not user:
        return jsonify(msg="Kullanıcı adı gerekli"), 400
    try:
        if db["users"].find_one({"user_id": user}):
            return jsonify(msg="Kullanıcı zaten mevcut"), 409 
        
        db["users"].insert_one({"user_id": user, "embeddings_enc": []})
        return jsonify(msg="Kullanıcı eklendi"), 200
    except Exception as e:
        print(f"MongoDB'ye kullanıcı eklerken hata: {e}")
        return jsonify(msg=f"Kullanıcı eklenirken veritabanı hatası oluştu: {str(e)}"), 500

@app.route("/users/<string:user_id>", methods=["DELETE"])
@jwt_required()
def delete_user(user_id):
    if db is None:
        return jsonify(msg="Veritabanı bağlantısı kurulamadı."), 500

    try:
        result = db["users"].delete_one({"user_id": user_id})
        if result.deleted_count > 0:
            return jsonify(msg=f"{user_id} silindi"), 200
        else:
            return jsonify(msg=f"{user_id} bulunamadı"), 404
    except Exception as e:
        print(f"MongoDB'den kullanıcı silerken hata: {e}")
        return jsonify(msg=f"Kullanıcı silinirken veritabanı hatası oluştu: {str(e)}"), 500

@app.route("/users", methods=["GET"])
@jwt_required()
def get_users():
    if db is None:
        return jsonify(msg="Veritabanı bağlantısı kurulamadı."), 500

    try:
        users = list(db["users"].find({}, {"_id": 0}))
        return jsonify(users), 200
    except Exception as e:
        print(f"MongoDB'den kullanıcıları çekerken hata: {e}")
        return jsonify(msg=f"Kullanıcılar alınırken veritabanı hatası oluştu: {str(e)}"), 500

@app.route("/users/<string:user_id>/embeddings", methods=["POST"])
@jwt_required()
def add_user_embeddings(user_id):
    if db is None:
        return jsonify(msg="Veritabanı bağlantısı kurulamadı."), 500

    data = request.json
    new_embeddings = data.get("embeddings")
    if not new_embeddings:
        return jsonify(msg="Embeddingler gerekli"), 400
    
    try:
        result = db["users"].update_one(
            {"user_id": user_id},
            {"$set": {"embeddings_enc": new_embeddings}}
        )
        if result.matched_count == 0:
            return jsonify(msg="Kullanıcı bulunamadı"), 404
        return jsonify(msg=f"{len(new_embeddings)} embedding {user_id} kullanıcısına eklendi"), 200
    except Exception as e:
        print(f"MongoDB'ye embedding eklerken hata: {e}")
        return jsonify(msg=f"Embedding eklenirken veritabanı hatası oluştu: {str(e)}"), 500

@app.route("/logs", methods=["GET"])
@jwt_required()
def get_logs():
    if db is None:
        return jsonify(msg="Veritabanı bağlantısı kurulamadı."), 500
    try:
        logs = list(db["failed_logins"].find({}, {"_id": 0}))
        return jsonify(logs), 200
    except Exception as e:
        print(f"MongoDB'den logları çekerken hata: {e}")
        return jsonify(msg=f"Loglar alınırken veritabanı hatası oluştu: {str(e)}"), 500

@app.route("/log_failed", methods=["POST"])
def log_failed():
    if db is None:
        return jsonify(msg="Veritabanı bağlantısı kurulamadı."), 500
    ip = request.remote_addr
    try:
        db["failed_logins"].insert_one({
            "timestamp": datetime.utcnow(),
            "ip": ip,
            "reason": "Yüz tanınmadı"
        })
        return jsonify(msg="Log kaydedildi"), 201
    except Exception as e:
        print(f"MongoDB'ye log kaydederken hata: {e}")
        return jsonify(msg=f"Log kaydedilirken veritabanı hatası oluştu: {str(e)}"), 500

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=5000)