Fsoky
4 years ago
7 changed files with 716 additions and 224 deletions
@ -0,0 +1,11 @@ |
|||
# Ignore pycache |
|||
|
|||
__pycache__ |
|||
|
|||
# Ignore dist |
|||
|
|||
/dist |
|||
|
|||
# Ignore pyproject |
|||
|
|||
pyproject.toml |
|||
@ -1,224 +0,0 @@ |
|||
import requests |
|||
import json |
|||
from flask import request |
|||
from websocket import create_connection |
|||
import socketio |
|||
|
|||
|
|||
class Scopes: |
|||
"""Удобный класс, для простого указания прав в приложении""" |
|||
|
|||
USER_SHOW = "oauth-user-show" |
|||
|
|||
DONATION_SUBSCRIBE = "oauth-donation-subscribe" |
|||
DONATION_INDEX = "oauth-donation-index" |
|||
|
|||
CUSTOM_ALERT_STORE = "oauth-custom_alert-store" |
|||
|
|||
GOAL_SUBSCRIBE = "oauth-goal-subscribe" |
|||
POLL_SUBSCRIBE = "oauth-poll-subscribe" |
|||
|
|||
ALL_SCOPES = [USER_SHOW, DONATION_INDEX, DONATION_SUBSCRIBE, CUSTOM_ALERT_STORE, |
|||
GOAL_SUBSCRIBE, POLL_SUBSCRIBE] |
|||
|
|||
|
|||
class Channels: |
|||
"""Права для подписки на каналы Centrifugo""" |
|||
|
|||
NEW_DONATION_ALERTS = "$alerts:donation_" |
|||
|
|||
DONATION_GOALS_UPDATES = "$goals:goal_" |
|||
|
|||
POLLS_UPDATES = "$polls:poll_" |
|||
|
|||
ALL_CHANNELS = [NEW_DONATION_ALERTS, DONATION_GOALS_UPDATES, POLLS_UPDATES] |
|||
|
|||
|
|||
class DonationAlertsApi: |
|||
"""Основной класс для работы с DA API""" |
|||
|
|||
def __init__(self, client_id, client_secret, redirect_uri, scopes): |
|||
symbols = [",", ", ", " ", "%20"] |
|||
|
|||
if isinstance(scopes, list): |
|||
obj_scopes = [] |
|||
for scope in scopes: |
|||
obj_scopes.append(scope) |
|||
|
|||
scopes = " ".join(obj_scopes) |
|||
|
|||
for symbol in symbols: |
|||
if symbol in scopes: |
|||
self.scope = scopes.replace(symbol, "%20").strip() # Replaces some symbols on '%20' for stable work |
|||
else: |
|||
self.scope = scopes |
|||
|
|||
self.client_id = client_id |
|||
self.client_secret = client_secret |
|||
self.redirect_uri = redirect_uri |
|||
self.login_url = f"https://www.donationalerts.com/oauth/authorize?client_id={client_id}&redirect_uri={redirect_uri}&response_type=code&scope={self.scope}" |
|||
self.token_url = f"https://www.donationalerts.com/oauth/token" |
|||
|
|||
# API LINKS |
|||
self.user_api = "https://www.donationalerts.com/api/v1/user/oauth" |
|||
self.donations_api = "https://www.donationalerts.com/api/v1/alerts/donations" |
|||
self.custom_alerts_api = "https://www.donationalerts.com/api/v1/custom_alert" |
|||
|
|||
def login(self): |
|||
return self.login_url |
|||
|
|||
def get_code(self): |
|||
return request.args.get("code") # Получаем аргмент "code" из адресной строки |
|||
|
|||
def get_access_token(self, code, full=False): |
|||
""" Параметр full=False |
|||
|
|||
Если True, то выводит весь json объект, а не только access_token |
|||
|
|||
""" |
|||
|
|||
payload = { |
|||
"client_id": self.client_id, |
|||
"client_secret": self.client_secret, |
|||
"grant_type": "authorization_code", |
|||
"code": code, |
|||
"redirect_uri": self.redirect_uri, |
|||
"scope": self.scope |
|||
} |
|||
|
|||
access_token = requests.post(url=self.token_url, data=payload).json() |
|||
self.refresh_token = access_token.get("refresh_token") |
|||
|
|||
return access_token if full else access_token.get("access_token") |
|||
|
|||
def get_donations(self, access_token): |
|||
headers = { |
|||
"Authorization": f"Bearer {access_token}", |
|||
"Content-Type": "application/x-www-form-urlencoded" |
|||
} |
|||
donate_object = requests.get(url=self.donations_api, headers=headers).json() |
|||
|
|||
return donate_object |
|||
|
|||
def get_user(self, access_token): |
|||
headers = { |
|||
"Authorization": f"Bearer {access_token}", |
|||
"Content-Type": "application/x-www-form-urlencoded" |
|||
} |
|||
user_object = requests.get(url=self.user_api, headers=headers).json() |
|||
|
|||
return user_object["data"] |
|||
|
|||
def send_custom_alert(self, access_token, external_id, headline, message, image_url=None, sound_url=None, is_shown=0): |
|||
headers = { |
|||
"Authorization": f"Bearer {access_token}", |
|||
"Content-Type": "application/x-www-form-urlencoded" |
|||
} |
|||
data = { |
|||
"external_id": external_id, |
|||
"header": headline, |
|||
"message": message, |
|||
"is_shown": is_shown, |
|||
"image_url": image_url, |
|||
"sound_url": sound_url |
|||
} |
|||
custom_alert_object = requests.post(url=self.custom_alerts_api, data=data, headers=headers).json() |
|||
|
|||
return custom_alert_object |
|||
|
|||
def get_refresh_token(self): |
|||
headers = {"Content-Type": "application/x-www-form-urlencoded"} |
|||
data = { |
|||
"grant_type": "refresh_token", |
|||
"client_id": self.client_id, |
|||
"client_secret": self.client_secret, |
|||
"refresh_token": self.refresh_token, |
|||
"redirect_uri": self.redirect_uri, |
|||
"scope": self.scope |
|||
} |
|||
|
|||
response = requests.post(url=self.token_url, data=data, headers=headers).json() |
|||
return response |
|||
|
|||
|
|||
class Centrifugo: |
|||
"""Получение событий в реальном времени (Oauth2)""" |
|||
|
|||
def __init__(self, socket_connection_token, access_token, user_id): |
|||
self.socket_connection_token = socket_connection_token |
|||
self.access_token = access_token |
|||
self.user_id = user_id |
|||
|
|||
self.uri = "wss://centrifugo.donationalerts.com/connection/websocket" |
|||
|
|||
def connect(self): |
|||
self.ws = create_connection(self.uri) # Подключаемся к серверу |
|||
self.ws.send(json.dumps( |
|||
{ |
|||
"params": { |
|||
"token": self.socket_connection_token |
|||
}, |
|||
"id": self.user_id |
|||
} |
|||
)) |
|||
self.ws_response = json.loads(self.ws.recv()) |
|||
|
|||
return self.ws_response |
|||
|
|||
def subscribe(self, channels): |
|||
chnls = [f"{channels}{self.user_id}"] |
|||
|
|||
if isinstance(channels, list): |
|||
chnls = [] |
|||
for channel in channels: |
|||
chnls.append(f"{channel}{self.user_id}") |
|||
|
|||
headers = { |
|||
"Authorization": f"Bearer {self.access_token}", |
|||
"Content-Type": "application/json" |
|||
} |
|||
data = { |
|||
"channels": chnls, |
|||
"client": self.ws_response["result"]["client"] |
|||
} |
|||
|
|||
response = requests.post(url="https://www.donationalerts.com/api/v1/centrifuge/subscribe", data=json.dumps(data), headers=headers).json() |
|||
for ch in response["channels"]: |
|||
self.ws.send(json.dumps( |
|||
{ |
|||
"params": { |
|||
"channel": ch["channel"], |
|||
"token": ch["token"] |
|||
}, |
|||
"method": 1, |
|||
"id": self.user_id |
|||
} |
|||
)) |
|||
|
|||
answer = {"response": self.ws.recv(), "sec_response": self.ws.recv()} |
|||
return answer # Возвращаем первые два ответа от сервера |
|||
|
|||
def listen(self): |
|||
return json.loads(self.ws.recv())["result"]["data"]["data"] # Возвращаем 3-й ответ от сервера |
|||
|
|||
sio = socketio.Client() |
|||
|
|||
|
|||
class Alert: |
|||
"""Получение донатов в реальном времени без Oauth2""" |
|||
|
|||
def __init__(self, token): |
|||
self.token = token # TOKEN можно скопировать здесь - https://www.donationalerts.com/dashboard/general |
|||
|
|||
def event(self): |
|||
def wrapper(function): |
|||
@sio.on("connect") |
|||
def on_connect(): |
|||
sio.emit("add-user", {"token": self.token, "type": "alert_widget"}) |
|||
|
|||
@sio.on("donation") |
|||
def on_message(data): |
|||
function(json.loads(data)) # Отправляем полученные данные в функцию |
|||
|
|||
sio.connect("wss://socket.donationalerts.ru:443", transports="websocket") |
|||
return wrapper |
|||
@ -0,0 +1,2 @@ |
|||
from .utils import * |
|||
from .donationalerts_api import * |
|||
@ -0,0 +1,300 @@ |
|||
import json |
|||
|
|||
import asyncio |
|||
import socketio |
|||
import aiohttp |
|||
import websockets |
|||
|
|||
from .utils import Event, User, Data, Donations, DonationsData, CentrifugoResponse |
|||
|
|||
DEFAULT_URL = "https://www.donationalerts.com/oauth/" |
|||
DEFAULT_API_LINK = "https://www.donationalerts.com/api/v1/" |
|||
|
|||
|
|||
class DonationAlertsApi: |
|||
|
|||
def __init__(self, client_id, client_secret, redirect_uri, scopes): |
|||
symbols = [",", ", ", " ", "%20"] |
|||
|
|||
if isinstance(scopes, list): |
|||
obj_scopes = [] |
|||
for scope in scopes: |
|||
obj_scopes.append(scope) |
|||
|
|||
scopes = " ".join(obj_scopes) |
|||
|
|||
for symbol in symbols: |
|||
if symbol in scopes: |
|||
self.scope = scopes.replace(symbol, "%20").strip() # Replaces some symbols on '%20' for stable work |
|||
else: |
|||
self.scope = scopes |
|||
|
|||
self.client_id = client_id |
|||
self.client_secret = client_secret |
|||
self.redirect_uri = redirect_uri |
|||
|
|||
def login(self): |
|||
return f"{DEFAULT_URL}authorize?client_id={self.client_id}&redirect_uri={self.redirect_uri}&response_type=code&scope={self.scope}" |
|||
|
|||
async def get_access_token(self, code, *, full_json=False): |
|||
payload = { |
|||
"client_id": self.client_id, |
|||
"client_secret": self.client_secret, |
|||
"grant_type": "authorization_code", |
|||
"code": code, |
|||
"redirect_uri": self.redirect_uri, |
|||
"scope": self.scope |
|||
} |
|||
|
|||
async with aiohttp.ClientSession() as session: |
|||
async with session.post(f"{DEFAULT_URL}token", data=payload) as response: |
|||
obj = await response.json() |
|||
|
|||
return Data( |
|||
obj["access_token"], |
|||
obj["expires_in"], |
|||
obj["refresh_token"], |
|||
obj["token_type"], |
|||
obj |
|||
) if full_json else obj["access_token"] |
|||
|
|||
async def donations_list(self, access_token, *, current_page: int=1, from_page: int=1, last_page: int=13, per_page: int=30, to: int=30, total: int=385): |
|||
headers = { |
|||
"Authorization": f"Bearer {access_token}", |
|||
"Content-Type": "application/x-www-form-urlencoded" |
|||
} |
|||
|
|||
"""Pagination in development""" |
|||
meta = { |
|||
"current_page": current_page, |
|||
"from": from_page, |
|||
"last_page": last_page, |
|||
"path": f"{DEFAULT_API_LINK}alerts/donations", |
|||
"per_page": per_page, |
|||
"to": to, |
|||
"total": total |
|||
} |
|||
links = { |
|||
"first": f"{DEFAULT_API_LINK}alerts/donations?page={from_page}", |
|||
"last": f"{DEFAULT_API_LINK}alerts/donations?page={last_page}", |
|||
"last": f"{DEFAULT_API_LINK}alerts/donations?page={last_page}", |
|||
"next": f"{DEFAULT_API_LINK}alerts/donations?page={current_page + 1}", |
|||
"prev": None |
|||
} |
|||
|
|||
async with aiohttp.ClientSession() as session: |
|||
async with session.get(f"{DEFAULT_API_LINK}alerts/donations", headers=headers) as response: |
|||
objs = await response.json() |
|||
donations = Donations(objects=objs["data"]) |
|||
|
|||
for obj in objs["data"]: |
|||
donation_object = DonationsData( |
|||
obj["amount"], |
|||
obj["amount_in_user_currency"], |
|||
obj["created_at"], |
|||
obj["currency"], |
|||
obj["id"], |
|||
obj["is_shown"], |
|||
obj["message"], |
|||
obj["message_type"], |
|||
obj["name"], |
|||
obj["payin_system"], |
|||
obj["recipient_name"], |
|||
obj["shown_at"], |
|||
obj["username"] |
|||
) |
|||
donations.donation.append(donation_object) |
|||
|
|||
return donations |
|||
|
|||
async def user(self, access_token): |
|||
headers = { |
|||
"Authorization": f"Bearer {access_token}", |
|||
"Content-Type": "application/x-www-form-urlencoded" |
|||
} |
|||
|
|||
async with aiohttp.ClientSession() as session: |
|||
async with session.get(f"{DEFAULT_API_LINK}user/oauth", headers=headers) as response: |
|||
obj = await response.json() |
|||
|
|||
return User( |
|||
obj["data"]["avatar"], |
|||
obj["data"]["code"], |
|||
obj["data"]["email"], |
|||
obj["data"]["id"], |
|||
obj["data"]["language"], |
|||
obj["data"]["name"], |
|||
obj["data"]["socket_connection_token"], |
|||
obj["data"] |
|||
) |
|||
|
|||
async def send_custom_alert(self, access_token, external_id, headline, message, *, image_url=None, sound_url=None, is_shown=0): |
|||
headers = { |
|||
"Authorization": f"Bearer {access_token}", |
|||
"Content-Type": "application/x-www-form-urlencoded" |
|||
} |
|||
payload = { |
|||
"external_id": external_id, |
|||
"headline": headline, |
|||
"message": message, |
|||
"is_shown": is_shown, |
|||
"image_url": image_url, |
|||
"sound_url": sound_url |
|||
} |
|||
|
|||
async with aiohttp.ClientSession() as session: |
|||
async with session.post(f"{DEFAULT_API_LINK}custom_alert", data=payload, headers=headers) as response: |
|||
return await response.json() |
|||
|
|||
async def get_refresh_token(self, access_token, refresh_token): |
|||
headers = { |
|||
"Content-Type": "application/x-www-form-urlencoded" |
|||
} |
|||
payload = { |
|||
"grant_type": "refresh_token", |
|||
"client_id": self.client_id, |
|||
"client_secret": self.client_secret, |
|||
"refresh_token": refresh_token, |
|||
"redirect_uri": self.redirect_uri, |
|||
"scope": self.scope |
|||
} |
|||
|
|||
async with aiohttp.ClientSession() as session: |
|||
async with session.post(f"{DEFAULT_URL}token", data=payload, headers=headers) as response: |
|||
obj = await response.json() |
|||
|
|||
return Data( |
|||
obj["access_token"], |
|||
obj["expires_in"], |
|||
obj["refresh_token"], |
|||
obj["token_type"], |
|||
obj |
|||
) |
|||
|
|||
|
|||
class Centrifugo: |
|||
|
|||
def __init__(self, socket_connection_token, access_token, user_id): |
|||
self.socket_connection_token = socket_connection_token |
|||
self.access_token = access_token |
|||
self.user_id = user_id |
|||
|
|||
async def subscribe(self, channels): |
|||
chnls = [f"{channels}{self.user_id}"] |
|||
if isinstance(channels, list): |
|||
chnls = [] |
|||
for channel in channels: |
|||
chnls.append(f"{channel}{self.user_id}") |
|||
|
|||
async with websockets.connect("wss://centrifugo.donationalerts.com/connection/websocket") as websocket: |
|||
await websocket.send(json.dumps( |
|||
{ |
|||
"params": { |
|||
"token": self.socket_connection_token |
|||
}, |
|||
"id": self.user_id |
|||
} |
|||
)) |
|||
|
|||
websocket_response = json.loads(await websocket.recv()) |
|||
|
|||
headers = { |
|||
"Authorization": f"Bearer {self.access_token}", |
|||
"Content-Type": "application/json" |
|||
} |
|||
data = { |
|||
"channels": chnls, |
|||
"client": websocket_response["result"]["client"] |
|||
} |
|||
|
|||
async with aiohttp.ClientSession(headers=headers) as session: |
|||
async with session.post(f"{DEFAULT_API_LINK}centrifuge/subscribe", data=json.dumps(data)) as response: |
|||
response = await response.json() |
|||
|
|||
for ch in response["channels"]: |
|||
await websocket.send(json.dumps( |
|||
{ |
|||
"params": { |
|||
"channel": ch["channel"], |
|||
"token": ch["token"] |
|||
}, |
|||
"method": 1, |
|||
"id": self.user_id |
|||
} |
|||
)) |
|||
|
|||
await websocket.recv() |
|||
await websocket.recv() |
|||
|
|||
obj = json.loads(await websocket.recv())["result"]["data"]["data"] |
|||
return CentrifugoResponse( |
|||
obj["amount"], |
|||
obj["amount_in_user_currency"], |
|||
obj["created_at"], |
|||
obj["currency"], |
|||
obj["id"], |
|||
obj["is_shown"], |
|||
obj["message"], |
|||
obj["message_type"], |
|||
obj["name"], |
|||
obj["payin_system"], |
|||
obj["recipient_name"], |
|||
obj["shown_at"], |
|||
obj["username"], |
|||
obj["reason"], |
|||
obj |
|||
) |
|||
|
|||
|
|||
sio = socketio.AsyncClient() |
|||
|
|||
|
|||
class Alert: |
|||
|
|||
def __init__(self, token): |
|||
self.token = token |
|||
|
|||
def event(self): |
|||
def decorate(function): |
|||
async def wrapper(): |
|||
|
|||
@sio.on("connect") |
|||
async def on_connect(): |
|||
await sio.emit("add-user", {"token": self.token, "type": "alert_widget"}) |
|||
|
|||
@sio.on("donation") |
|||
async def on_message(data): |
|||
data = json.loads(data) |
|||
|
|||
await function( |
|||
Event( |
|||
data["id"], |
|||
data["alert_type"], |
|||
data["is_shown"], |
|||
json.loads(data["additional_data"]), |
|||
data["billing_system"], |
|||
data["billing_system_type"], |
|||
data["username"], |
|||
data["amount"], |
|||
data["amount_formatted"], |
|||
data["amount_main"], |
|||
data["currency"], |
|||
data["message"], |
|||
data["header"], |
|||
data["date_created"], |
|||
data["emotes"], |
|||
data["ap_id"], |
|||
data["_is_test_alert"], |
|||
data["message_type"], |
|||
data["preset_id"], |
|||
data |
|||
) |
|||
) |
|||
|
|||
await sio.connect("wss://socket.donationalerts.ru:443", transports="websocket") |
|||
|
|||
loop = asyncio.get_event_loop() |
|||
loop.run_until_complete(wrapper()) |
|||
return loop.run_forever() |
|||
|
|||
return decorate |
|||
@ -0,0 +1,286 @@ |
|||
import json |
|||
import requests |
|||
|
|||
from websocket import create_connection |
|||
import socketio |
|||
|
|||
from .utils import Event, User, Data, Donations, DonationsData, CentrifugoResponse |
|||
|
|||
DEFAULT_URL = "https://www.donationalerts.com/oauth/" |
|||
DEFAULT_API_LINK = "https://www.donationalerts.com/api/v1/" |
|||
|
|||
|
|||
class DonationAlertsApi: |
|||
""" |
|||
This class describes work with Donation Alerts API |
|||
""" |
|||
|
|||
def __init__(self, client_id, client_secret, redirect_uri, scopes): |
|||
symbols = [",", ", ", " ", "%20"] |
|||
|
|||
if isinstance(scopes, list): |
|||
obj_scopes = [] |
|||
for scope in scopes: |
|||
obj_scopes.append(scope) |
|||
|
|||
scopes = " ".join(obj_scopes) |
|||
|
|||
for symbol in symbols: |
|||
if symbol in scopes: |
|||
self.scope = scopes.replace(symbol, "%20").strip() # Replaces some symbols on '%20' for stable work |
|||
else: |
|||
self.scope = scopes |
|||
|
|||
self.client_id = client_id |
|||
self.client_secret = client_secret |
|||
self.redirect_uri = redirect_uri |
|||
|
|||
def login(self): |
|||
return f"{DEFAULT_URL}authorize?client_id={self.client_id}&redirect_uri={self.redirect_uri}&response_type=code&scope={self.scope}" |
|||
|
|||
def get_access_token(self, code, *, full_json=False): |
|||
payload = { |
|||
"client_id": self.client_id, |
|||
"client_secret": self.client_secret, |
|||
"grant_type": "authorization_code", |
|||
"code": code, |
|||
"redirect_uri": self.redirect_uri, |
|||
"scope": self.scope |
|||
} |
|||
|
|||
access_token = requests.post(f"{DEFAULT_URL}token", data=payload).json() |
|||
self.refresh_token = access_token.get("refresh_token") |
|||
|
|||
return Data( |
|||
obj["access_token"], |
|||
obj["expires_in"], |
|||
obj["refresh_token"], |
|||
obj["token_type"], |
|||
obj |
|||
) if full_json else obj["access_token"] |
|||
|
|||
def donations_list(self, access_token, *, current_page: int=1, from_page: int=1, last_page: int=13, per_page: int=30, to: int=30, total: int=385): |
|||
headers = { |
|||
"Authorization": f"Bearer {access_token}", |
|||
"Content-Type": "application/x-www-form-urlencoded" |
|||
} |
|||
|
|||
"""Pagination in development""" |
|||
meta = { |
|||
"current_page": current_page, |
|||
"from": from_page, |
|||
"last_page": last_page, |
|||
"path": f"{DEFAULT_API_LINK}alerts/donations", |
|||
"per_page": per_page, |
|||
"to": to, |
|||
"total": total |
|||
} |
|||
links = { |
|||
"first": f"{DEFAULT_API_LINK}alerts/donations?page={from_page}", |
|||
"last": f"{DEFAULT_API_LINK}alerts/donations?page={last_page}", |
|||
"last": f"{DEFAULT_API_LINK}alerts/donations?page={last_page}", |
|||
"next": f"{DEFAULT_API_LINK}alerts/donations?page={current_page + 1}", |
|||
"prev": None |
|||
} |
|||
|
|||
objs = requests.get(f"{DEFAULT_API_LINK}alerts/donations", headers=headers).json() |
|||
donations = Donations(objects=objs["data"]) |
|||
|
|||
for obj in objs["data"]: |
|||
donation_object = DonationsData( |
|||
obj["amount"], |
|||
obj["amount_in_user_currency"], |
|||
obj["created_at"], |
|||
obj["currency"], |
|||
obj["id"], |
|||
obj["is_shown"], |
|||
obj["message"], |
|||
obj["message_type"], |
|||
obj["name"], |
|||
obj["payin_system"], |
|||
obj["recipient_name"], |
|||
obj["shown_at"], |
|||
obj["username"] |
|||
) |
|||
donations.donation.append(donation_object) |
|||
|
|||
return donations |
|||
|
|||
def user(self, access_token): |
|||
headers = { |
|||
"Authorization": f"Bearer {access_token}", |
|||
"Content-Type": "application/x-www-form-urlencoded" |
|||
} |
|||
obj = requests.get(f"{DEFAULT_API_LINK}user/oauth", headers=headers).json() |
|||
|
|||
return User( |
|||
obj["data"]["avatar"], |
|||
obj["data"]["code"], |
|||
obj["data"]["email"], |
|||
obj["data"]["id"], |
|||
obj["data"]["language"], |
|||
obj["data"]["name"], |
|||
obj["data"]["socket_connection_token"], |
|||
obj["data"] |
|||
) |
|||
|
|||
def send_custom_alert(self, access_token, external_id, headline, message, *, image_url=None, sound_url=None, is_shown=0): |
|||
headers = { |
|||
"Authorization": f"Bearer {access_token}", |
|||
"Content-Type": "application/x-www-form-urlencoded" |
|||
} |
|||
payload = { |
|||
"external_id": external_id, |
|||
"headline": headline, |
|||
"message": message, |
|||
"is_shown": is_shown, |
|||
"image_url": image_url, |
|||
"sound_url": sound_url |
|||
} |
|||
|
|||
obj = requests.post(f"{DEFAULT_API_LINK}custom_alert", data=payload, headers=headers).json() |
|||
return obj |
|||
|
|||
def get_refresh_token(self, access_token, refresh_token): |
|||
headers = { |
|||
"Content-Type": "application/x-www-form-urlencoded" |
|||
} |
|||
payload = { |
|||
"grant_type": "refresh_token", |
|||
"client_id": self.client_id, |
|||
"client_secret": self.client_secret, |
|||
"refresh_token": refresh_token, |
|||
"redirect_uri": self.redirect_uri, |
|||
"scope": self.scope |
|||
} |
|||
|
|||
obj = requests.post(f"{DEFAULT_URL}token", data=payload, headers=headers).json() |
|||
return Data( |
|||
obj["access_token"], |
|||
obj["expires_in"], |
|||
obj["refresh_token"], |
|||
obj["token_type"], |
|||
obj |
|||
) |
|||
|
|||
|
|||
class Centrifugo: |
|||
|
|||
def __init__(self, socket_connection_token, access_token, user_id): |
|||
self.socket_connection_token = socket_connection_token |
|||
self.access_token = access_token |
|||
self.user_id = user_id |
|||
|
|||
self.uri = "wss://centrifugo.donationalerts.com/connection/websocket" |
|||
|
|||
def subscribe(self, channels): |
|||
chnls = [f"{channels}{self.user_id}"] |
|||
if isinstance(channels, list): |
|||
chnls = [] |
|||
for channel in channels: |
|||
chnls.append(f"{channel}{self.user_id}") |
|||
|
|||
ws = create_connection(self.uri) |
|||
ws.send(json.dumps( |
|||
{ |
|||
"params": { |
|||
"token": self.socket_connection_token |
|||
}, |
|||
"id": self.user_id |
|||
} |
|||
)) |
|||
|
|||
ws_response = json.loads(ws.recv()) |
|||
|
|||
headers = { |
|||
"Authorization": f"Bearer {self.access_token}", |
|||
"Content-Type": "application/json" |
|||
} |
|||
data = { |
|||
"channels": chnls, |
|||
"client": ws_response["result"]["client"] |
|||
} |
|||
|
|||
response = requests.post(f"{DEFAULT_API_LINK}centrifuge/subscribe", data=json.dumps(data), headers=headers).json() |
|||
for ch in response["channels"]: |
|||
self.ws.send(json.dumps( |
|||
{ |
|||
"params": { |
|||
"channel": ch["channel"], |
|||
"token": ch["token"] |
|||
}, |
|||
"method": 1, |
|||
"id": self.user_id |
|||
} |
|||
)) |
|||
|
|||
ws.recv() |
|||
ws.recv() |
|||
|
|||
obj = json.loads(ws.recv())["result"]["data"]["data"] |
|||
return CentrifugoResponse( |
|||
obj["amount"], |
|||
obj["amount_in_user_currency"], |
|||
obj["created_at"], |
|||
obj["currency"], |
|||
obj["id"], |
|||
obj["is_shown"], |
|||
obj["message"], |
|||
obj["message_type"], |
|||
obj["name"], |
|||
obj["payin_system"], |
|||
obj["recipient_name"], |
|||
obj["shown_at"], |
|||
obj["username"], |
|||
obj["reason"], |
|||
obj |
|||
) |
|||
|
|||
|
|||
sio = socketio.Client() |
|||
|
|||
|
|||
class Alert: |
|||
|
|||
def __init__(self, token): |
|||
self.token = token |
|||
|
|||
def event(self): |
|||
def wrapper(function): |
|||
|
|||
@sio.on("connect") |
|||
def on_connect(): |
|||
sio.emit("add-user", {"token": self.token, "type": "alert_widget"}) |
|||
|
|||
@sio.on("donation") |
|||
def on_message(data): |
|||
data = json.loads(data) |
|||
|
|||
function( |
|||
Event( |
|||
data["id"], |
|||
data["alert_type"], |
|||
data["is_shown"], |
|||
json.loads(data["additional_data"]), |
|||
data["billing_system"], |
|||
data["billing_system_type"], |
|||
data["username"], |
|||
data["amount"], |
|||
data["amount_formatted"], |
|||
data["amount_main"], |
|||
data["currency"], |
|||
data["message"], |
|||
data["header"], |
|||
data["date_created"], |
|||
data["emotes"], |
|||
data["ap_id"], |
|||
data["_is_test_alert"], |
|||
data["message_type"], |
|||
data["preset_id"], |
|||
data |
|||
) |
|||
) |
|||
|
|||
sio.connect("wss://socket.donationalerts.ru:443", transports="websocket") |
|||
|
|||
return wrapper |
|||
@ -0,0 +1,23 @@ |
|||
class Scopes: |
|||
USER_SHOW = "oauth-user-show" |
|||
|
|||
DONATION_SUBSCRIBE = "oauth-donation-subscribe" |
|||
DONATION_INDEX = "oauth-donation-index" |
|||
|
|||
CUSTOM_ALERT_STORE = "oauth-custom_alert-store" |
|||
|
|||
GOAL_SUBSCRIBE = "oauth-goal-subscribe" |
|||
POLL_SUBSCRIBE = "oauth-poll-subscribe" |
|||
|
|||
ALL_SCOPES = [USER_SHOW, DONATION_INDEX, DONATION_SUBSCRIBE, CUSTOM_ALERT_STORE, |
|||
GOAL_SUBSCRIBE, POLL_SUBSCRIBE] |
|||
|
|||
|
|||
class Channels: |
|||
NEW_DONATION_ALERTS = "$alerts:donation_" |
|||
|
|||
DONATION_GOALS_UPDATES = "$goals:goal_" |
|||
|
|||
POLLS_UPDATES = "$polls:poll_" |
|||
|
|||
ALL_CHANNELS = [NEW_DONATION_ALERTS, DONATION_GOALS_UPDATES, POLLS_UPDATES] |
|||
@ -0,0 +1,94 @@ |
|||
from dataclasses import dataclass |
|||
|
|||
|
|||
@dataclass |
|||
class Event: |
|||
|
|||
id: int |
|||
alert_type: str |
|||
is_shown: str |
|||
additional_data: dict |
|||
billing_system: str |
|||
billing_system_type: str |
|||
username: str |
|||
amount: str |
|||
amount_formatted: str |
|||
amount_main: int |
|||
currency: str |
|||
message: str |
|||
header: str |
|||
date_created: str |
|||
emotes: str |
|||
ap_id: str |
|||
_is_test_alert: bool |
|||
message_type: str |
|||
preset_id: int |
|||
objects: dict |
|||
|
|||
|
|||
@dataclass |
|||
class Donations: |
|||
|
|||
donation = [] |
|||
objects: dict = None |
|||
|
|||
|
|||
@dataclass |
|||
class DonationsData: |
|||
|
|||
amount: int |
|||
amount_in_user_currency: float |
|||
created_at: str |
|||
currency: str |
|||
id: int |
|||
is_shown: int |
|||
message: str |
|||
message_type: str |
|||
name: str |
|||
payin_system: str |
|||
recipient_name: str |
|||
shown_at: str |
|||
username: str |
|||
|
|||
|
|||
@dataclass |
|||
class User: |
|||
|
|||
avatar: str |
|||
code: str |
|||
email: str |
|||
id: int |
|||
language: str |
|||
name: str |
|||
socket_connection_token: str |
|||
objects: dict |
|||
|
|||
|
|||
@dataclass |
|||
class Data: |
|||
|
|||
access_token: str |
|||
expires_in: int |
|||
refresh_token: str |
|||
token_type: str |
|||
objects: dict |
|||
|
|||
|
|||
@dataclass |
|||
class CentrifugoResponse: |
|||
|
|||
amount: int |
|||
amount_in_user_currency: float |
|||
created_at: str |
|||
currency: str |
|||
id: int |
|||
is_shown: int |
|||
message: str |
|||
message_type: str |
|||
name: str |
|||
payin_system: str |
|||
recipient_name: str |
|||
shown_at: str |
|||
username: str |
|||
reason: str |
|||
objects: dict |
|||
Loading…
Reference in new issue