104 lines
3.8 KiB
Python
104 lines
3.8 KiB
Python
import aiohttp
|
||
import asyncio
|
||
import json
|
||
|
||
class V2rayAController:
|
||
def __init__(self, host='http://localhost:2017', username='admin', password='admin'):
|
||
self.host = host
|
||
self.username = username
|
||
self.password = password
|
||
self.token = None
|
||
self.session = None
|
||
|
||
async def authenticate(self):
|
||
"""Аутентификация и получение токена"""
|
||
self.session = aiohttp.ClientSession()
|
||
auth_url = f"{self.host}/api/login"
|
||
credentials = {
|
||
'username': self.username,
|
||
'password': self.password
|
||
}
|
||
|
||
try:
|
||
async with self.session.post(auth_url, json=credentials) as response:
|
||
if response.status == 200:
|
||
data = await response.json()
|
||
self.token = data.get('data').get('token')
|
||
print("V2RayA auth OK")
|
||
return True
|
||
else:
|
||
print(f"Ошибка аутентификации: {response.status}")
|
||
return False
|
||
except Exception as e:
|
||
print(f"Ошибка при аутентификации: {e}")
|
||
return False
|
||
|
||
async def enable_proxy(self):
|
||
"""Включение прокси"""
|
||
if not self.token:
|
||
print("Не выполнен вход. Пожалуйста, сначала выполните аутентификацию")
|
||
return False
|
||
|
||
url = f"{self.host}/api/v2ray"
|
||
headers = {'Authorization': f"Bearer {self.token}"}
|
||
data = {}
|
||
|
||
try:
|
||
async with self.session.post(url, headers=headers, json=data) as response:
|
||
if response.status == 200:
|
||
print("Прокси успешно включен")
|
||
return True
|
||
else:
|
||
print(f"Ошибка при включении прокси: {response.status}")
|
||
return False
|
||
except Exception as e:
|
||
print(f"Ошибка при включении прокси: {e}")
|
||
return False
|
||
|
||
async def disable_proxy(self):
|
||
"""Выключение прокси"""
|
||
if not self.token:
|
||
print("Не выполнен вход. Пожалуйста, сначала выполните аутентификацию")
|
||
return False
|
||
|
||
url = f"{self.host}/api/v2ray"
|
||
headers = {'Authorization': f"Bearer {self.token}"}
|
||
data = {'operation': 'stop'}
|
||
|
||
try:
|
||
async with self.session.delete(url, headers=headers, json=data) as response:
|
||
if response.status == 200:
|
||
print("Прокси успешно выключен")
|
||
return True
|
||
else:
|
||
print(f"Ошибка при выключении прокси: {response.status}")
|
||
return False
|
||
except Exception as e:
|
||
print(f"Ошибка при выключении прокси: {e}")
|
||
return False
|
||
|
||
async def close(self):
|
||
"""Закрытие сессии"""
|
||
if self.session:
|
||
await self.session.close()
|
||
|
||
async def main():
|
||
# Инициализация контроллера
|
||
controller = V2rayAController(
|
||
host='http://localhost:2017',
|
||
username='admin123',
|
||
password='admin123'
|
||
)
|
||
|
||
# Аутентификация
|
||
if await controller.authenticate():
|
||
# Примеры использования
|
||
await controller.enable_proxy()
|
||
await asyncio.sleep(20) # Пауза для демонстрации
|
||
await controller.disable_proxy()
|
||
|
||
# Закрытие сессии
|
||
await controller.close()
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(main()) |