Update app.py

app.py

@@ -21,6 +21,14 @@ import uuid
 import psutil
 
 
+
+import decimal
+import hashlib
+from urllib import parse
+from urllib.parse import urlparse
+
+
+
 from base import replace_null_with_empty_string
 
 
@@ -940,6 +948,171 @@ def add_user_senler_full():
 
 
 
+
+
+
+
+
+
+# ROBO
+def calculate_signature(*args) -> str:
+    """Create signature MD5."""
+    return hashlib.md5(':'.join(str(arg) for arg in args).encode()).hexdigest()
+
+def parse_response(request: str) -> dict:
+    """
+    :param request: Link.
+    :return: Dictionary.
+    """
+    params = {}
+    for item in urlparse(request).query.split('&'):
+        key, value = item.split('=')
+        params[key] = value
+    return params
+
+def check_signature_result(
+    order_number: int,  # invoice number
+    received_sum: decimal,  # cost of goods, RU
+    received_signature: hex,  # SignatureValue
+    password: str  # Merchant password
+) -> bool:
+    signature = calculate_signature(received_sum, order_number, password)
+    return signature.lower() == received_signature.lower()
+
+def generate_payment_link(
+    merchant_login: str,  # Merchant login
+    merchant_password_1: str,  # Merchant password
+    cost: decimal,  # Cost of goods, RU
+    number: int,  # Invoice number
+    description: str,  # Description of the purchase
+    user_ip: str,  # User IP address
+    is_test: str,  # Test mode flag
+    robokassa_payment_url = 'https://auth.robokassa.ru/Merchant/Index.aspx',
+) -> str:
+    """URL for redirection of the customer to the service."""
+    signature = calculate_signature(
+        merchant_login,
+        cost,
+        number,
+        user_ip,
+        merchant_password_1
+    )
+
+    data = {
+        'MerchantLogin': merchant_login,
+        'OutSum': cost,
+        'InvId': number,
+        'Description': description,
+        'SignatureValue': signature,
+        'IsTest': is_test,
+        'UserIp': user_ip
+    }
+    return f'{robokassa_payment_url}?{parse.urlencode(data)}'
+
+def update_rkassa(email, rkassa_data):
+    for db in DATABASES:
+        conn = sqlite3.connect(db)
+        cursor = conn.cursor()
+
+        # Получаем текущие данные из столбца rkassa
+        cursor.execute('SELECT rkassa FROM contacts WHERE email = ?', (email,))
+        current_rkassa = cursor.fetchone()
+
+        if current_rkassa and current_rkassa[0]:
+            # Если данные уже есть, добавляем новые через запятую
+            new_rkassa = f"{current_rkassa[0]},{rkassa_data}"
+        else:
+            # Если данных нет, просто записываем новые
+            new_rkassa = rkassa_data
+
+        # Обновляем столбец rkassa
+        cursor.execute('UPDATE contacts SET rkassa = ? WHERE email = ?', (new_rkassa, email))
+        conn.commit()
+        conn.close()
+
+@app.route('/set_invid', methods=['GET'])
+def set_invid():
+    global invid
+    invid = int(request.args.get('invid'))
+    return jsonify({'status': 'success', 'invid': invid})
+
+@app.route('/process_payment', methods=['POST'])
+def process_payment():
+    global invid
+
+    # Получаем данные из формы
+    out_sum = request.form.get('out_sum', '10.00')
+    user_name = request.form.get('name', '')
+    user_email = request.form.get('email', '')
+    user_phone = request.form.get('phone', '')
+    user_ip = request.form.get('user_ip', '')
+    user_is_test = request.form.get('user_is_test', '0')
+
+    product_name = request.form.get('product_name', 'product')
+    product_id = request.form.get('product_id', 'default_product_id')
+    expiration_date = request.form.get('expiration_date', '2024-01-16T12:00')
+    dostup_date = request.form.get('dostup_date', '')
+
+    # Проверяем, что invid установлен
+    if invid is None:
+        return jsonify({'status': 'error', 'message': 'invid is not set'})
+
+    # Увеличиваем invid на один
+    invid += 1
+
+    # Формируем данные для запроса к Robokassa
+    payment_link = generate_payment_link(
+        merchant_login,
+        password_1,
+        decimal.Decimal(out_sum),
+        invid,
+        f"Оплата за {product_name}",
+        user_ip,
+        user_is_test
+    )
+
+    # Отправляем запрос на сервер Robokassa
+    response = requests.get(payment_link)
+
+    # Записываем данные в базу данных
+    rkassa_data = json.dumps({str(invid): {"data_dostup": dostup_date, "product": product_id}})
+    update_rkassa(user_email, rkassa_data)
+
+    # Записываем текущее время в столбец data_on
+    utc_now = datetime.utcnow()
+    msk_tz = pytz.timezone('Europe/Moscow')
+    msk_now = utc_now.replace(tzinfo=pytz.utc).astimezone(msk_tz)
+    data_on = msk_now.strftime('%Y-%m-%d %H:%M:%S')
+
+    for db in DATABASES:
+        conn = sqlite3.connect(db)
+        cursor = conn.cursor()
+        cursor.execute('UPDATE contacts SET data_on = ?, name = ?, phone = ? WHERE email = ?', (data_on, user_name, user_phone, user_email))
+        conn.commit()
+        conn.close()
+
+    # Возвращаем ссылку на страницу оплаты
+    return jsonify({'payment_url': payment_link})
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
 
 
 # Поднятие страницы с таблицей