-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
241 lines (215 loc) · 11.4 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
import os
from dotenv import load_dotenv
import discord
from discord.ext import commands
from discord import app_commands
import traceback
import util
import plot
import keep_alive
import requests
import time
import json
load_dotenv()
backend_url = os.getenv('BACKEND_URL')
api_password = os.getenv('API_PASSWORD')
token = os.getenv('TOKEN')
app_id = os.getenv('APP_ID')
guild_id = os.getenv('GUILD_ID')
KNS = discord.Object(id=guild_id)
class menuView(discord.ui.View):
def __init__(self):
super().__init__(timeout=None)
bitgetRegisterBtn = discord.ui.Button(label='註冊Bitget', style=discord.ButtonStyle.grey, url='https://partner.bitget.com/bg/2bqn83471678523358144')
notionLinkBtn = discord.ui.Button(label='參數設置教學', style=discord.ButtonStyle.grey, url='https://r-chieh.notion.site/KNS-Bitget-4221105143ed41779eefcf29dc044783')
self.add_item(bitgetRegisterBtn)
self.add_item(notionLinkBtn)
@discord.ui.button(label='跟單收益試算', style=discord.ButtonStyle.green, custom_id='persistent_view:green')
async def green(self, interaction: discord.Interaction, button: discord.ui.Button):
await interaction.response.send_modal(simulate())
@discord.ui.button(label='跟單風險試算', style=discord.ButtonStyle.red, custom_id='persistent_view:red')
async def red(self, interaction: discord.Interaction, button: discord.ui.Button):
await interaction.response.send_modal(analyze())
@discord.ui.button(label='連線狀態測試', style=discord.ButtonStyle.grey, custom_id='persistent_view:grey')
async def grey(self, interaction: discord.Interaction, button: discord.ui.Button):
await interaction.response.send_message("I'm ready!", ephemeral=True)
class simulate(discord.ui.Modal, title='交易員實盤績效查詢&試算'):
traderName = discord.ui.TextInput(
label = 'bitget交易員名稱',
placeholder = '輸入帶單員名稱'
)
margin = discord.ui.TextInput(
label = '開倉大小',
placeholder = '每倉保證金（USDT）ex:10'
)
lossPerPos = discord.ui.TextInput(
label = '倉位止損',
placeholder = '自訂倉位止損（％）ex:100'
)
startDate = discord.ui.TextInput(
label = '試算起始時間',
placeholder = '輸入『西元年-月份-日期』ex:2023-02-01'
)
endDate = discord.ui.TextInput(
label = '試算結束時間',
placeholder = '輸入『西元年-月份-日期』ex:2023-04-01'
)
async def on_submit(self, interaction: discord.Interaction):
await interaction.response.send_message("計算中...", ephemeral = True)
userName = interaction.user
util.log(userName, f'績效試算({str(self.traderName)}, {str(self.margin)}, {str(self.lossPerPos)}, {str(self.startDate)}, {str(self.endDate)})')
# resData = await util.copySimulate(str(self.traderName), int(str(self.margin)), int(str(self.lossPerPos)), str(self.startDate))
payload = {
"trader": str(self.traderName),
"startDate": str(self.startDate),
"endDate": str(self.endDate),
"margin": int(str(self.margin)),
"stoploss": int(str(self.lossPerPos)),
}
header = {
"password": api_password,
}
response = requests.post(url=f'{backend_url}/calculate/DC/copySimulate', headers=header, json=payload)
resData = response.json()
resData = resData["data"]
# print(f'type: {type(resData)}')
# print(resData.keys())
if not resData:
await interaction.edit_original_response(content="查無交易員資料")
util.log(userName, f'查無交易員資料')
else:
filename = plot.plotLine(resData["chartX"], resData["chartY"], str(self.startDate), str(self.endDate))
file = discord.File(f'./{ filename }', filename="image.png")
embed = discord.Embed(
title = f'{ str(self.startDate) } -> { util.dateNow() } 跟單試算',
type = "rich",
description = f'交易員: {resData["traderName"]}'
)\
.add_field(name=" ", value=" ", inline=False)\
.add_field(name="本金（每倉）", value=f'{ str(self.margin) }USDT')\
.add_field(name="倉位", value=f'共{ resData["positionCount"] }倉')\
.add_field(name=f"{ str(self.lossPerPos) }％止損", value=f'共觸發{ resData["countSL"] }次')\
.add_field(name="總收益（無爆倉）", value=f'{ resData["copyProfit"] }USDT', inline=False)\
.set_image(url="attachment://image.png")
await interaction.edit_original_response(content="計算完成：", attachments = [file], embed = embed)
if os.path.exists(f'{ filename }'):
os.remove(f'{ filename }')
print(f'{ filename } deleted.')
else:
print("The file does not exist!")
async def on_error(self, interaction: discord.Interaction, error: Exception) -> None:
await interaction.edit_original_response(content='發生錯誤，請稍候再試一次。')
# Make sure we know what the error actually is
traceback.print_exception(type(error), error, error.__traceback__)
class analyze(discord.ui.Modal, title='交易員風險與回撤試算'):
traderName = discord.ui.TextInput(
label = 'bitget交易員名稱',
placeholder = '輸入帶單員名稱'
)
initialCapital = discord.ui.TextInput(
label = '本金',
placeholder = '總資金（USDT）ex:10000'
)
maxLossPercent = discord.ui.TextInput(
label = '最大回撤',
placeholder = '可承受之最大回撤（％）ex:20'
)
startDate = discord.ui.TextInput(
label = '試算起始時間',
placeholder = '輸入『西元年-月份-日期』ex:2023-02-01'
)
endDate = discord.ui.TextInput(
label = '試算結束時間',
placeholder = '輸入『西元年-月份-日期』ex:2023-04-01'
)
async def on_submit(self, interaction: discord.Interaction):
await interaction.response.send_message("計算中...", ephemeral = True)
userName = interaction.user
util.log(userName, f'風險試算({str(self.traderName)}, {str(self.initialCapital)}, {str(self.maxLossPercent)}, {str(self.startDate)}, {str(self.endDate)})')
resData = await util.analyzeTraderMDD(str(self.traderName), int(str(self.initialCapital)), int(str(self.maxLossPercent)), str(self.startDate))
payload = {
"trader": str(self.traderName),
"startDate": str(self.startDate),
"endDate": str(self.endDate),
"capital": str(self.initialCapital),
"maxLoss": str(self.maxLossPercent),
}
header = {
"password": api_password,
}
response = requests.post(url=f'{backend_url}/calculate/DC/analyzeMDD', headers=header, json=payload)
resData = response.json()
resData = resData["data"]
# print(f'type: {type(resData)}')
# print(resData.keys())
if not resData:
await interaction.edit_original_response(content="查無交易員資料")
util.log(userName, f'查無交易員資料')
else:
# print("交易員：" + resData["traderName"])
period = f'{ str(self.startDate) } -> { util.dateNow() }'
embed = discord.Embed(
title = f'{ period } 風險試算',
type = "rich",
description = f'交易員: {resData["traderName"]}'
)\
.add_field(name=" ", value=" ", inline=False)\
.add_field(name="本金", value=f'{ str(self.initialCapital) }USDT')\
.add_field(name="設定最大回撤", value=f'{ str(self.maxLossPercent) }％')\
.add_field(name="試算時間段", value=f'{ period }')\
.add_field(name=" ", value=" ", inline=False)\
.add_field(name="最大回合持倉數", value=f'{ resData["maxPosition"] }')\
.add_field(name="最大浮動虧損 (倉)", value=f'{ resData["drawdownHighest"] }倍倉')\
.add_field(name="最大浮動虧損 (％)", value=f'{ resData["drawdownHighestPercent"] }％')\
.add_field(name=" ", value=" ", inline=False)\
.add_field(name="發生最大浮動虧損時持倉數", value=f'{ resData["drawdownHighestPosition"] }')\
.add_field(name="潛在極端浮動虧損 (倉)", value=f'{ resData["potentialMDD"] }倍倉')\
.add_field(name="潛在極端浮動虧損 (％)", value=f'{ resData["potentialMDDPercent"] }％')\
.add_field(name=" ", value=" ", inline=False)\
.add_field(name="風報比平均值", value=f'{ resData["RDMean"] }')\
.add_field(name="風報比標準差", value=f'{ resData["RDDev"] }')\
.add_field(name="風報比中位數", value=f'{ resData["RDMid"] }')\
.add_field(name=" ", value=" ", inline=False)\
.add_field(name="建議安全保證金", value=f'{ resData["safeMargin"] }')\
.add_field(name="建議安全倉數上限", value=f'{ resData["positionEstimate"] }')\
.add_field(name=" ", value=" ")\
.add_field(name="※潛在極端", value="將前 N 大單筆保證金浮動虧損組合進最大回合持倉內計算", inline=False)\
.add_field(name="※建議安全保證金", value="(最大可接受虧損金額 ÷ 最大保證金浮動虧損 (％)) ÷ 最大保證金浮動虧損時持倉數", inline=False)\
.add_field(name="※建議安全總倉數", value="持倉數平均值 + 2 × 持倉數標準差 (包含 95% 事件)", inline=False)\
.add_field(name="※風報比", value="收益 ÷ 最大浮動虧損", inline=False)
await interaction.edit_original_response(content="計算完成：", embed = embed)
async def on_error(self, interaction: discord.Interaction, error: Exception) -> None:
await interaction.edit_original_response(content='發生錯誤，請稍候再試一次。')
# Make sure we know what the error actually is
traceback.print_exception(type(error), error, error.__traceback__)
class CopyBot(discord.Client):
def __init__(self):
intents = discord.Intents.default()
intents.members = True
intents.message_content = True
intents.guilds = True
super().__init__(command_prefix=commands.when_mentioned_or('/'), intents=intents)
self.tree = app_commands.CommandTree(self)
async def setup_hook(self) -> None:
await self.tree.sync(guild = KNS)
self.add_view(menuView())
async def on_ready(self):
print(f'Logged in as {self.user} (ID: {self.user.id})')
mainChannel = self.get_channel(1085904328319045683)
botLogChannel = self.get_channel(944846007051620404)
msg = await mainChannel.fetch_message(1085907306849521684)
await msg.edit(content="Bitget 交易員分析 & 試算工具", view=menuView())
# await mainChannel.send("Bitget 交易員分析 & 試算工具", view=menuView())
await botLogChannel.send("I'm ready!")
print('ready------')
bot = CopyBot()
@bot.tree.command(name = "ping", description = "pong!", guild = KNS)
async def ping(ctx):
await ctx.response.send_message("pong!")
keep_alive.keep_alive()
import os
try:
bot.run(token)
except Exception as e:
print(e)
os.system("kill 1")