blackanbar/utils.py at main · salmanultra/blackanbar · GitHub

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
import pandas as pd
import os
from datetime import datetime
from models import Product, Transaction, User, DataManager

# توابع کمکی برای مدیریت فایل‌های Excel

def load_products_from_excel(data_manager: DataManager, file_path: str = 'products.xlsx'):
    """بارگذاری کالاها از فایل Excel"""
    if os.path.exists(file_path):
        df = pd.read_excel(file_path)
        for _, row in df.iterrows():
            product = Product(
                code=row['کد'],
                name=row['نام'],
                category=row['دسته‌بندی'],
                capacity=int(row['ظرفیت']),
                current_stock=int(row['موجودی فعلی'])
            )
            data_manager.add_product(product)
    else:
        print(f"فایل {file_path} یافت نشد. فایل جدید ایجاد خواهد شد.")

def save_products_to_excel(data_manager: DataManager, file_path: str = 'products.xlsx'):
    """ذخیره کالاها به فایل Excel"""
    data = []
    for product in data_manager.get_products():
        data.append({
            'کد': product.code,
            'نام': product.name,
            'دسته‌بندی': product.category,
            'ظرفیت': product.capacity,
            'موجودی فعلی': product.current_stock
        })
    df = pd.DataFrame(data)
    df.to_excel(file_path, index=False)

def load_transactions_from_excel(data_manager: DataManager, file_path: str = 'transactions.xlsx'):
    """بارگذاری تراکنش‌ها از فایل Excel"""
    if os.path.exists(file_path):
        df = pd.read_excel(file_path)
        for _, row in df.iterrows():
            transaction = Transaction(
                product_code=row['کد محصول'],
                transaction_type=row['نوع تراکنش'],
                quantity=int(row['مقدار']),
                date=pd.to_datetime(row['تاریخ']),
                user=row['کاربر']
            )
            data_manager.add_transaction(transaction)
    else:
        print(f"فایل {file_path} یافت نشد. فایل جدید ایجاد خواهد شد.")

def save_transactions_to_excel(data_manager: DataManager, file_path: str = 'transactions.xlsx'):
    """ذخیره تراکنش‌ها به فایل Excel"""
    data = []
    for transaction in data_manager.get_transactions():
        data.append({
            'کد محصول': transaction.product_code,
            'نوع تراکنش': transaction.transaction_type,
            'مقدار': transaction.quantity,
            'تاریخ': transaction.date,
            'کاربر': transaction.user
        })
    df = pd.DataFrame(data)
    df.to_excel(file_path, index=False)

def load_users_from_excel(data_manager: DataManager, file_path: str = 'users.xlsx'):
    """بارگذاری کاربران از فایل Excel"""
    if os.path.exists(file_path):
        df = pd.read_excel(file_path)
        for _, row in df.iterrows():
            user = User(
                username=row['نام کاربری'],
                password=row['رمز عبور'],
                role=row['نقش']
            )
            data_manager.add_user(user)
    else:
        print(f"فایل {file_path} یافت نشد. فایل جدید ایجاد خواهد شد.")

def save_users_to_excel(data_manager: DataManager, file_path: str = 'users.xlsx'):
    """ذخیره کاربران به فایل Excel"""
    data = []
    for user in data_manager.get_users():
        data.append({
            'نام کاربری': user.username,
            'رمز عبور': user.password,
            'نقش': user.role
        })
    df = pd.DataFrame(data)
    df.to_excel(file_path, index=False)

def validate_product(product: Product) -> bool:
    """اعتبارسنجی داده‌های کالا"""
    if not product.code or not product.name or product.capacity < 0 or product.current_stock < 0:
        return False
    return True

def validate_transaction(transaction: Transaction, products: list) -> bool:
    """اعتبارسنجی تراکنش"""
    product_codes = [p.code for p in products]
    if transaction.product_code not in product_codes or transaction.quantity <= 0 or transaction.transaction_type not in ['ورود', 'خروج']:
        return False
    return True