dev@locazia: add ton_daemon

This commit is contained in:
user 2024-03-01 23:44:09 +03:00
parent f0a0350a57
commit 0c271e3482
6 changed files with 132 additions and 0 deletions

View File

@ -83,6 +83,8 @@ if __name__ == '__main__':
from app.core.background.indexator_service import main as target_fn
elif startup_target == 'uploader':
from app.core.background.uploader_service import main as target_fn
elif startup_target == 'ton_daemon':
from app.core.background.ton_service import main as target_fn
startup_fn = startup_fn or target_fn
assert startup_fn

View File

@ -0,0 +1,79 @@
from base64 import b64encode
from httpx import AsyncClient
from app.core.logger import make_log
from app.core._config import TONCENTER_HOST, TONCENTER_API_KEY
import asyncio
import time
class TonCenter:
def __init__(self, host: str, api_key: str = None, testnet: bool = False):
self.host = host
self.api_key = api_key
self.last_used = time.time()
async def request(self, method: str, endpoint: str, *args, **kwargs) -> dict:
async with AsyncClient(headers={
'Authorization': f"Bearer {self.api_key}"
} if self.api_key else None) as client:
if self.api_key:
kwargs['headers'] = kwargs.get('headers', {})
kwargs['headers']['X-API-Key'] = self.api_key
if not self.api_key:
while time.time() < self.last_used + 1.2:
await asyncio.sleep(0.1)
self.last_used = time.time()
response = await client.request(method, f"{self.host}{endpoint}", *args, **kwargs)
if response.status_code != 200:
raise Exception(f'Error while TONCENTER request {endpoint}: {response.text}')
try:
return response.json()
except BaseException as e:
make_log("Toncenter", f'Error while request {endpoint}: {e}' + '\n' + response.text, level='error')
return await self.request(method, endpoint, *args, **kwargs)
async def send_boc(self, src: bytes):
try:
res = await self.request(
"POST", 'sendBoc',
json={'boc': b64encode(src).decode()}
)
except Exception as e:
make_log("Toncenter", f"sendBoc error: {e}", level='error')
return None
make_log("Toncenter", f"sendBoc: {res}", level='debug')
return res
async def get_account(self, addr: str):
return (await self.request(
"GET", 'getAddressInformation',
params={'address': addr}
)).get('result', {})
async def get_seqno(self, addr: str):
return (await self.request(
"GET", 'getWalletInformation',
params={'address': addr}
)).get('result', {}).get('seqno', 0)
async def get_transactions(self, addr: str, limit: int = 10):
return (await self.request(
"GET", 'getTransactions',
params={'address': addr, 'limit': limit}
)).get('result', [])
async def run_get_method(self, addr, method, stack=[]):
return (await self.request(
"POST", 'runGetMethod', json={
'address': addr,
'method': method,
'stack': stack if type(stack) == list else []
}
)).get('result', {})
toncenter = TonCenter(TONCENTER_HOST, TONCENTER_API_KEY)

View File

@ -36,3 +36,6 @@ ALLOWED_CONTENT_TYPES = [
'video/mp4', 'video/webm', 'video/ogg',
'audio/mpeg', 'audio/ogg', 'audio/wav',
]
TONCENTER_HOST = os.getenv('TONCENTER_HOST', 'https://toncenter.com/api/v1/')
TONCENTER_API_KEY = os.getenv('TONCENTER_API_KEY')

View File

@ -0,0 +1,29 @@
from app.core.logger import make_log
from app.core.storage import db_session
from app.core._secrets import service_wallet
from app.core._blockchain.ton.toncenter import toncenter
import asyncio
async def main():
make_log("TON", "Service started", level="info")
sw_seqno_result = await toncenter.run_get_method(service_wallet.address.to_string(1, 1, 1), 'seqno')
make_log("TON", f"Service wallet run seqno method: {sw_seqno_result}", level="info")
while True:
make_log("TON", "Service running", level="debug")
# with db_session() as session:
# for stored_content in session.query(StoredContent).filter(StoredContent.uploaded == False).all():
# pass
await asyncio.sleep(5)
if __name__ == '__main__':
loop = asyncio.get_event_loop()
loop.run_until_complete(main())
loop.close()

View File

@ -66,3 +66,21 @@ services:
depends_on:
maria_db:
condition: service_healthy
ton_daemon:
build:
context: .
dockerfile: Dockerfile
command: python -m app ton_daemon
env_file:
- .env
links:
- maria_db
volumes:
- ./logs:/app/logs
- ./storedContent:/app/data
- ./activeConfig:/app/config
depends_on:
maria_db:
condition: service_healthy

View File

@ -7,3 +7,4 @@ aiogram==3.4.1
pytonconnect==0.3.0
base58==2.1.1
tonsdk==1.0.13
httpx==0.25.0