Update database.py

database.py

@@ -1,65 +1,121 @@
+import json
+import os
 import boto3
 from pymongo import MongoClient
-import os
+from dotenv import load_dotenv
+
+load_dotenv()
 
-# AWS DynamoDB ve MongoDB bağlantıları için gerekli ayarlar
+# Çevresel değişkenlerden yapılandırma bilgilerini alıyoruz
+DYNAMO_TABLE = os.getenv('DYNAMO_TABLE')
+MONGO_URI = os.getenv('MONGO_URI')
+INPUT_DB = os.getenv('INPUT_DB')
+
+# AWS DynamoDB kaynağı
 dynamodb = boto3.resource('dynamodb')
-MONGO_URI = os.getenv('MONGO_URI', 'mongodb://localhost:27017')
-REFERENCE_DB = os.getenv('REFERENCE_DB', 'EgitimDatabase')
-INPUT_DB = os.getenv('INPUT_DB', 'InputDatabase')
-DYNAMO_TABLE = os.getenv('DYNAMO_TABLE', 'UserInputs')
+
+def lambda_handler(event, context):
+    """AWS Lambda işleyici fonksiyonu."""
+    method = event['requestContext']['http']['method']
+    
+    if method == 'POST':
+        return handle_post(event)
+    elif method == 'GET':
+        return handle_get(event)
+    else:
+        return {
+            'statusCode': 405,
+            'body': json.dumps('Method Not Allowed')
+        }
+
+def handle_post(event):
+    """POST isteklerini işleyin."""
+    table = dynamodb.Table(DYNAMO_TABLE)
+    
+    try:
+        body = json.loads(event['body'])
+        title = body.get('title')
+        keywords = body.get('keywords')
+        text = body.get('text')
+        
+        if not title or not keywords or not text:
+            return {
+                'statusCode': 400,
+                'body': json.dumps('Invalid input data')
+            }
+        
+        # DynamoDB'ye veri ekleyin
+        table.put_item(
+            Item={
+                'title': title,
+                'keywords': keywords,
+                'text': text
+            }
+        )
+        
+        # MongoDB'ye veri ekleyin
+        insert_data_into_input_db([{"title": title, "keywords": keywords, "text": text}])
+        
+        return {
+            'statusCode': 200,
+            'body': json.dumps('Data inserted into DynamoDB and MongoDB!')
+        }
+    
+    except Exception as e:
+        return {
+            'statusCode': 500,
+            'body': json.dumps(f"Error: {str(e)}")
+        }
+
+def handle_get(event):
+    """GET isteklerini işleyin."""
+    table = dynamodb.Table(DYNAMO_TABLE)
+    
+    try:
+        query_params = event.get('queryStringParameters', {})
+        title = query_params.get('title')
+        
+        if not title:
+            return {
+                'statusCode': 400,
+                'body': json.dumps('Title parameter is required')
+            }
+        
+        # DynamoDB'den veri alın
+        response = table.get_item(
+            Key={
+                'title': title
+            }
+        )
+        
+        item = response.get('Item', {})
+        
+        if not item:
+            return {
+                'statusCode': 404,
+                'body': json.dumps('Data not found')
+            }
+        
+        return {
+            'statusCode': 200,
+            'body': json.dumps(item)
+        }
+    
+    except Exception as e:
+        return {
+            'statusCode': 500,
+            'body': json.dumps(f"Error: {str(e)}")
+        }
 
 def connect_to_mongodb():
     """MongoDB'ye bağlan."""
     client = MongoClient(MONGO_URI)
     return client
 
-def get_reference_data():
-    """Reference database'den verileri al."""
-    client = connect_to_mongodb()
-    db = client[REFERENCE_DB]
-    collection = db['test']  #test koleksiyonu içerisinde 
-    data = collection.find({}, {"title": 1, "keywords": 1, "text": 1, "_id": 0})
-    return list(data)
-
 def insert_data_into_input_db(data):
     """Input database'e veri ekle."""
     client = connect_to_mongodb()
     db = client[INPUT_DB]
-    collection = db['input']  #kullanıcıdan alınacak verilerin saklandığı koleksiyon adı 
+    collection = db['input']  # Kullanıcıdan alınacak verilerin saklandığı koleksiyon
     collection.insert_many(data)
-
-def insert_data_into_dynamodb(data):
-    """Kullanıcıdan alınan verileri DynamoDB'ye ekle."""
-    table = dynamodb.Table(input_to_users)
-    for item in data:
-        table.put_item(Item=item)
-
-def get_user_input():
-    """Kullanıcıdan veri al."""
-    # Lambda fonksiyonu ile kullanıcıdan gelen veriyi alın
-    user_input = [
-        {"title": "Sample Title", "keywords": ["keyword1", "keyword2","keyword3","keyword4","keyword5"], "text": "Sample text."}
-    ]
-    return user_input
-
-def main():
-    """Ana fonksiyon."""
-    client = connect_to_mongodb()
-
-    # Referans veriyi MongoDB'den alın
-    reference_data = get_reference_data()
-
-    # Kullanıcıdan veri alın (Lambda aracılığıyla alınacak)
-    user_data = get_user_input()
-
-    # Kullanıcı verilerini DynamoDB'ye ekle
-    insert_data_into_dynamodb(user_data)
-
-    # Kullanıcı verilerini MongoDB'deki input database'e ekle
-    insert_data_into_input_db(user_data)
-    
     client.close()
-
-if __name__ == "__main__":
-    main()