File size: 13,180 Bytes
a854c1b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
d9e7bdd
 
 
 
 
 
 
 
 
 
 
 
 
a854c1b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
d9e7bdd
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
"""
موصل قاعدة البيانات لنظام إدارة المناقصات
"""

import os
import sqlite3
import logging

logger = logging.getLogger('tender_system.database')

class DatabaseConnector:
    """فئة موصل قاعدة البيانات"""
    
    def __init__(self, config):
        """تهيئة موصل قاعدة البيانات"""
        self.config = config
        self.db_config = config.get_database_config()
        self.db_path = self.db_config.get('path')
        self.connection = None
        self.cursor = None
        
        # إنشاء قاعدة البيانات إذا لم تكن موجودة
        self._initialize_database()
    
    def _initialize_database(self):
        """تهيئة قاعدة البيانات"""
        try:
            # التأكد من وجود المجلد
            os.makedirs(os.path.dirname(self.db_path), exist_ok=True)
            
            # إنشاء الاتصال
            self.connection = sqlite3.connect(self.db_path)
            self.cursor = self.connection.cursor()
            
            # إنشاء الجداول إذا لم تكن موجودة
            self._create_tables()
            
            # إضافة بيانات افتراضية إذا كانت قاعدة البيانات فارغة
            self._add_default_data()
            
            logger.info(f"تم تهيئة قاعدة البيانات بنجاح: {self.db_path}")
        except Exception as e:
            logger.error(f"خطأ في تهيئة قاعدة البيانات: {str(e)}")
            raise
    
    def _create_tables(self):
        """إنشاء جداول قاعدة البيانات"""
        # جدول المشاريع المحفوظة
        self.cursor.execute('''
        CREATE TABLE IF NOT EXISTS saved_projects (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            project_name TEXT NOT NULL,
            project_code TEXT,
            project_description TEXT,
            boq_data TEXT NOT NULL,
            total_cost REAL NOT NULL,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
        )
        ''')

        # جدول المستخدمين
        self.cursor.execute('''
        CREATE TABLE IF NOT EXISTS users (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            username TEXT UNIQUE NOT NULL,
            password TEXT NOT NULL,
            full_name TEXT NOT NULL,
            email TEXT UNIQUE NOT NULL,
            role TEXT NOT NULL,
            status TEXT NOT NULL,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
        )
        ''')
        
        # جدول المشاريع
        self.cursor.execute('''
        CREATE TABLE IF NOT EXISTS projects (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT NOT NULL,
            client TEXT NOT NULL,
            description TEXT,
            start_date TEXT,
            end_date TEXT,
            status TEXT NOT NULL,
            created_by INTEGER,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (created_by) REFERENCES users (id)
        )
        ''')
        
        # جدول المستندات
        self.cursor.execute('''
        CREATE TABLE IF NOT EXISTS documents (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            project_id INTEGER,
            name TEXT NOT NULL,
            file_path TEXT NOT NULL,
            document_type TEXT NOT NULL,
            description TEXT,
            uploaded_by INTEGER,
            uploaded_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (project_id) REFERENCES projects (id),
            FOREIGN KEY (uploaded_by) REFERENCES users (id)
        )
        ''')
        
        # جدول بنود التسعير
        self.cursor.execute('''
        CREATE TABLE IF NOT EXISTS pricing_items (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            project_id INTEGER,
            item_number TEXT NOT NULL,
            description TEXT NOT NULL,
            unit TEXT NOT NULL,
            quantity REAL NOT NULL,
            unit_price REAL NOT NULL,
            total_price REAL NOT NULL,
            created_by INTEGER,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (project_id) REFERENCES projects (id),
            FOREIGN KEY (created_by) REFERENCES users (id)
        )
        ''')
        
        # جدول الموارد البشرية
        self.cursor.execute('''
        CREATE TABLE IF NOT EXISTS human_resources (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT NOT NULL,
            position TEXT NOT NULL,
            daily_cost REAL NOT NULL,
            skills TEXT,
            status TEXT NOT NULL,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
        )
        ''')
        
        # جدول المعدات
        self.cursor.execute('''
        CREATE TABLE IF NOT EXISTS equipment (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT NOT NULL,
            type TEXT NOT NULL,
            daily_cost REAL NOT NULL,
            status TEXT NOT NULL,
            location TEXT,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
        )
        ''')
        
        # جدول المواد
        self.cursor.execute('''
        CREATE TABLE IF NOT EXISTS materials (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT NOT NULL,
            unit TEXT NOT NULL,
            quantity REAL NOT NULL,
            unit_price REAL NOT NULL,
            supplier TEXT,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
        )
        ''')
        
        # جدول المخاطر
        self.cursor.execute('''
        CREATE TABLE IF NOT EXISTS risks (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            project_id INTEGER,
            name TEXT NOT NULL,
            category TEXT NOT NULL,
            probability TEXT NOT NULL,
            impact TEXT NOT NULL,
            risk_level TEXT NOT NULL,
            mitigation_strategy TEXT,
            created_by INTEGER,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (project_id) REFERENCES projects (id),
            FOREIGN KEY (created_by) REFERENCES users (id)
        )
        ''')
        
        # جدول التقارير
        self.cursor.execute('''
        CREATE TABLE IF NOT EXISTS reports (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT NOT NULL,
            project_id INTEGER,
            report_type TEXT NOT NULL,
            period TEXT,
            file_path TEXT,
            created_by INTEGER,
            status TEXT NOT NULL,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (project_id) REFERENCES projects (id),
            FOREIGN KEY (created_by) REFERENCES users (id)
        )
        ''')
        
        # حفظ التغييرات
        self.connection.commit()
    
    def _add_default_data(self):
        """إضافة بيانات افتراضية"""
        # التحقق من وجود مستخدمين
        self.cursor.execute("SELECT COUNT(*) FROM users")
        user_count = self.cursor.fetchone()[0]
        
        if user_count == 0:
            # إضافة مستخدم افتراضي (admin/admin)
            self.cursor.execute('''
            INSERT INTO users (username, password, full_name, email, role, status)
            VALUES (?, ?, ?, ?, ?, ?)
            ''', ('admin', 'admin', 'مدير النظام', '[email protected]', 'مدير', 'نشط'))
            
            # إضافة مستخدمين إضافيين
            self.cursor.execute('''
            INSERT INTO users (username, password, full_name, email, role, status)
            VALUES (?, ?, ?, ?, ?, ?)
            ''', ('user1', 'password', 'أحمد محمد', '[email protected]', 'مستخدم', 'نشط'))
            
            self.cursor.execute('''
            INSERT INTO users (username, password, full_name, email, role, status)
            VALUES (?, ?, ?, ?, ?, ?)
            ''', ('user2', 'password', 'سارة أحمد', '[email protected]', 'مستخدم', 'نشط'))
            
            # حفظ التغييرات
            self.connection.commit()
            
            logger.info("تم إضافة بيانات المستخدمين الافتراضية")
        
        # التحقق من وجود مشاريع
        self.cursor.execute("SELECT COUNT(*) FROM projects")
        project_count = self.cursor.fetchone()[0]
        
        if project_count == 0:
            # إضافة مشاريع افتراضية
            self.cursor.execute('''
            INSERT INTO projects (name, client, description, start_date, end_date, status, created_by)
            VALUES (?, ?, ?, ?, ?, ?, ?)
            ''', ('مشروع تطوير الطريق السريع', 'وزارة النقل', 'مشروع تطوير وتوسعة الطريق السريع', '2025-01-15', '2025-12-31', 'نشط', 1))
            
            self.cursor.execute('''
            INSERT INTO projects (name, client, description, start_date, end_date, status, created_by)
            VALUES (?, ?, ?, ?, ?, ?, ?)
            ''', ('مشروع بناء المدرسة الثانوية', 'وزارة التعليم', 'مشروع بناء مدرسة ثانوية جديدة', '2025-02-01', '2025-08-30', 'نشط', 1))
            
            self.cursor.execute('''
            INSERT INTO projects (name, client, description, start_date, end_date, status, created_by)
            VALUES (?, ?, ?, ?, ?, ?, ?)
            ''', ('مشروع تجديد المستشفى', 'وزارة الصحة', 'مشروع تجديد وتطوير المستشفى', '2024-10-15', '2025-03-15', 'مكتمل', 1))
            
            # حفظ التغييرات
            self.connection.commit()
            
            logger.info("تم إضافة بيانات المشاريع الافتراضية")
    
    def execute_query(self, query, params=None):
        """تنفيذ استعلام"""
        try:
            if params:
                self.cursor.execute(query, params)
            else:
                self.cursor.execute(query)
            
            self.connection.commit()
            return self.cursor
        except Exception as e:
            logger.error(f"خطأ في تنفيذ الاستعلام: {str(e)}")
            self.connection.rollback()
            raise
    
    def fetch_one(self, query, params=None):
        """جلب صف واحد"""
        cursor = self.execute_query(query, params)
        return cursor.fetchone()
    
    def fetch_all(self, query, params=None):
        """جلب جميع الصفوف"""
        cursor = self.execute_query(query, params)
        return cursor.fetchall()
    
    def insert(self, table, data):
        """إدراج بيانات"""
        columns = ', '.join(data.keys())
        placeholders = ', '.join(['?' for _ in data])
        query = f"INSERT INTO {table} ({columns}) VALUES ({placeholders})"
        
        try:
            self.cursor.execute(query, list(data.values()))
            self.connection.commit()
            return self.cursor.lastrowid
        except Exception as e:
            logger.error(f"خطأ في إدراج البيانات: {str(e)}")
            self.connection.rollback()
            raise
    
    def update(self, table, data, condition):
        """تحديث بيانات"""
        set_clause = ', '.join([f"{column} = ?" for column in data.keys()])
        query = f"UPDATE {table} SET {set_clause} WHERE {condition}"
        
        try:
            self.cursor.execute(query, list(data.values()))
            self.connection.commit()
            return self.cursor.rowcount
        except Exception as e:
            logger.error(f"خطأ في تحديث البيانات: {str(e)}")
            self.connection.rollback()
            raise
    
    def delete(self, table, condition):
        """حذف بيانات"""
        query = f"DELETE FROM {table} WHERE {condition}"
        
        try:
            self.cursor.execute(query)
            self.connection.commit()
            return self.cursor.rowcount
        except Exception as e:
            logger.error(f"خطأ في حذف البيانات: {str(e)}")
            self.connection.rollback()
            raise
    
    def close(self):
        """إغلاق الاتصال"""
        if self.connection:
            self.connection.close()
            logger.info("تم إغلاق الاتصال بقاعدة البيانات")