Wahbi-AI / database /db_connector.py
EGYADMIN's picture
Upload 106 files
5a66745 verified
raw
history blame
1.23 kB
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
وحدة الاتصال بقاعدة البيانات
"""
import os
import sys
import psycopg2
from dotenv import load_dotenv
# إضافة مسار النظام للوصول للملفات المشتركة
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
# تحميل متغيرات البيئة
load_dotenv()
def get_connection():
"""
إنشاء اتصال بقاعدة البيانات
الإرجاع:
اتصال بقاعدة البيانات
"""
try:
# محاولة الاتصال بقاعدة البيانات
conn = psycopg2.connect(
dbname=os.getenv("PGDATABASE"),
user=os.getenv("PGUSER"),
password=os.getenv("PGPASSWORD"),
host=os.getenv("PGHOST"),
port=os.getenv("PGPORT")
)
return conn
except Exception as e:
print(f"خطأ في الاتصال بقاعدة البيانات: {e}")
# إذا فشل الاتصال، استخدم اتصال قاعدة بيانات وهمي
from utils.helpers import get_connection as get_mock_connection
return get_mock_connection()