247 lines
8.3 KiB
Python
247 lines
8.3 KiB
Python
# Qiwi module v1.10
|
||
# 17/05/2021
|
||
# https://t.me/ssleg © 2020 – 2021
|
||
|
||
import logging
|
||
from datetime import datetime, timedelta
|
||
from pathlib import Path
|
||
|
||
import requests
|
||
|
||
headers = {
|
||
'accept': 'application/json',
|
||
'content-type': 'application/json'
|
||
}
|
||
|
||
url = 'https://api.qiwi.com/partner/bill/v1/bills/'
|
||
|
||
error_flag = True
|
||
theme_code = None
|
||
|
||
|
||
# не подлежит прямому вызову.
|
||
# проверяет, что модуль инициализирован.
|
||
def init_check():
|
||
if error_flag:
|
||
levent = 'module is not initialized.'
|
||
logging.error(levent)
|
||
return False
|
||
return True
|
||
|
||
|
||
# не подлежит прямому вызову.
|
||
# проверяет корректность суммы счета.
|
||
def zero_check(amount):
|
||
if type(amount) == int or type(amount) == float:
|
||
if amount <= 0:
|
||
levent = 'bill amount is zero or negative.'
|
||
logging.error(levent)
|
||
return False
|
||
return True
|
||
else:
|
||
levent = 'bill amount is not a number. positive int or float values possible.'
|
||
logging.error(levent)
|
||
return False
|
||
|
||
|
||
# не подлежит прямому вызову.
|
||
# возврашает строку со временем действия счета в формате сервера Qiwi.
|
||
def get_valid_to_time(hours, mins):
|
||
if type(hours) == int and type(mins) == int:
|
||
if hours >= 0 and mins > 0:
|
||
now = datetime.now()
|
||
delta = timedelta(hours=hours, minutes=mins)
|
||
valid_to = now + delta
|
||
valid_to_string = str(valid_to)[0:19]
|
||
rezult_string = valid_to_string.replace(' ', 'T') + '+03:00'
|
||
return rezult_string
|
||
else:
|
||
levent = 'hours or minutes must be above zero.'
|
||
logging.error(levent)
|
||
return False
|
||
else:
|
||
levent = 'hours or minutes is not integer.'
|
||
logging.error(levent)
|
||
return False
|
||
|
||
|
||
# создание счета на оплату. при успехе возвращает url с формой оплаты для клиента.
|
||
# при неуспехе - False.
|
||
# примеры использования смотрите в sample.py и adv_sample.py
|
||
def create_bill(bill_amount, bill_id, comment_string=None, valid_hours=0, valid_mins=15):
|
||
if not init_check():
|
||
return False
|
||
|
||
if not zero_check(bill_amount):
|
||
return False
|
||
|
||
if type(bill_amount) == float:
|
||
full_penny = bill_amount * 100
|
||
fract = full_penny - int(full_penny)
|
||
if fract != 0:
|
||
levent = 'bill amount must have integer number of penny (kopeek).'
|
||
logging.error(levent)
|
||
return False
|
||
|
||
if type(bill_id) != str or bill_id == '':
|
||
levent = 'bill id is not a string or empty.'
|
||
logging.error(levent)
|
||
return False
|
||
|
||
valid_to = get_valid_to_time(valid_hours, valid_mins)
|
||
if not valid_to:
|
||
return False
|
||
|
||
amount = {'currency': 'RUB', 'value': '{:.2f}'.format(bill_amount)}
|
||
|
||
request_data = {'amount': amount, 'expirationDateTime': valid_to}
|
||
|
||
if comment_string is not None:
|
||
if type(comment_string) == str and comment_string != '':
|
||
request_data['comment'] = comment_string
|
||
else:
|
||
levent = 'comment is not a string or empty.'
|
||
logging.warning(levent)
|
||
|
||
if theme_code is not None:
|
||
custom = {'themeCode': theme_code}
|
||
request_data['customFields'] = custom
|
||
|
||
request_url = url + bill_id
|
||
|
||
try:
|
||
response = requests.put(request_url, json=request_data, headers=headers, timeout=5)
|
||
response_code = response.status_code
|
||
if response_code == 200:
|
||
response_dict = response.json()
|
||
return response_dict.get('payUrl')
|
||
elif response_code == 401:
|
||
levent = 'Qiwi autorization error. invalid secret key.'
|
||
logging.error(levent)
|
||
return False
|
||
else:
|
||
response_text = response.text
|
||
levent = 'Qiwi server error (create bill). code - ' + str(response_code) + ', response - ' + response_text
|
||
logging.error(levent)
|
||
return False
|
||
|
||
except Exception as e:
|
||
levent = 'protocol error (create bill): ' + str(e)
|
||
logging.error(levent)
|
||
return False
|
||
|
||
|
||
# проверка статуса счета,на входе его идентификатор (текст), на выходе один из 4х статусов, если успешно:
|
||
# 'WAITING' - cчет выставлен, ожидает оплаты.
|
||
# 'PAID' - cчет оплачен.
|
||
# 'REJECTED' - счет отменен.
|
||
# 'EXPIRED' - счет не оплачен и истек срок его действия.
|
||
# можно вызывать 1 раз в секунду и реже.
|
||
# если неуспешно - возвращает False.
|
||
def bill_status(bill_id):
|
||
if not init_check():
|
||
return False
|
||
|
||
if type(bill_id) != str or bill_id == '':
|
||
levent = 'bill id is not a string or empty.'
|
||
logging.error(levent)
|
||
return False
|
||
|
||
request_url = url + bill_id
|
||
|
||
try:
|
||
response = requests.get(request_url, headers=headers, timeout=5)
|
||
response_code = response.status_code
|
||
if response_code == 200:
|
||
response_dict = response.json()
|
||
status = response_dict.get('status')
|
||
return status.get('value')
|
||
elif response_code == 401:
|
||
levent = 'Qiwi autorization error. invalid secret key.'
|
||
logging.error(levent)
|
||
return False
|
||
else:
|
||
response_text = response.text
|
||
levent = 'Qiwi server error (bill status). code - ' + str(response_code) + ', response - ' + response_text
|
||
logging.error(levent)
|
||
return False
|
||
|
||
except Exception as e:
|
||
levent = 'protocol error (bill status): ' + str(e)
|
||
logging.error(levent)
|
||
return False
|
||
|
||
|
||
# отмена счета, на входе его идентификатор (текст).
|
||
# в случае успеха возвращает REJECTED, иначе False
|
||
def cancel_bill(bill_id):
|
||
if not init_check():
|
||
return False
|
||
|
||
if type(bill_id) != str or bill_id == '':
|
||
levent = 'bill id is not a string or empty.'
|
||
logging.error(levent)
|
||
return False
|
||
|
||
request_url = url + bill_id + '/reject'
|
||
|
||
try:
|
||
response = requests.post(request_url, headers=headers, timeout=5)
|
||
response_code = response.status_code
|
||
if response_code == 200:
|
||
response_dict = response.json()
|
||
status = response_dict.get('status')
|
||
return status.get('value')
|
||
elif response_code == 401:
|
||
levent = 'Qiwi autorization error. invalid secret key.'
|
||
logging.error(levent)
|
||
return False
|
||
else:
|
||
response_text = response.text
|
||
levent = 'Qiwi server error (cancel bill). code - ' + str(response_code) + ', response - ' + response_text
|
||
logging.error(levent)
|
||
return False
|
||
|
||
except Exception as e:
|
||
levent = 'protocol error (cancel bill): ' + str(e)
|
||
logging.error(levent)
|
||
return False
|
||
|
||
|
||
# инициализация модуля - загрузка секретного ключа из файла и настройка пользовательской темы для формы оплаты.
|
||
def init(theme=None):
|
||
global error_flag
|
||
global theme_code
|
||
|
||
if theme is not None:
|
||
if type(theme) == str and theme != '':
|
||
theme_code = theme
|
||
else:
|
||
levent = 'custom theme code is not a string or empty. theme not used.'
|
||
logging.warning(levent)
|
||
|
||
key_path = Path('qiwi_key.txt')
|
||
if not key_path.exists():
|
||
levent = 'Qiwi key file not found, module work is not possible. write secret key to file qiwi_key.txt'
|
||
logging.error(levent)
|
||
return False
|
||
|
||
file = open('qiwi_key.txt')
|
||
qiwi_key = file.readline()
|
||
file.close()
|
||
|
||
if qiwi_key.find('\n') > -1:
|
||
qiwi_key = qiwi_key[0:len(qiwi_key) - 1]
|
||
|
||
if not 180 < len(qiwi_key) < 230:
|
||
levent = 'Qiwi key not found in file, module work is not possible. write secret key to file qiwi_key.txt'
|
||
logging.error(levent)
|
||
return False
|
||
|
||
headers['Authorization'] = 'Bearer ' + qiwi_key
|
||
|
||
levent = 'Qiwi key loaded, init completed.'
|
||
logging.info(levent)
|
||
error_flag = False
|
||
return True
|