/
main.py
162 lines (142 loc) · 6.45 KB
/
main.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
import asyncio
import sys
from logging import getLogger, basicConfig
from pywebio import start_server
from pywebio.input import *
from pywebio.output import *
from pywebio.session import defer_call, get_current_task_id
from enums import WitchRule, GuardRule, Role, GameStage
from models.room import Room
from models.user import User
from utils import add_cancel_button, get_interface_ip
basicConfig(stream=sys.stdout, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
logger = getLogger('Wolf')
logger.setLevel('DEBUG')
async def main():
"""狼人杀"""
put_markdown("## 狼人杀法官")
current_user = User.alloc(
await input('请输入你的昵称',
required=True,
validate=User.validate_nick,
help_text='请使用一个易于分辨的名称'),
get_current_task_id()
)
@defer_call
def on_close():
User.free(current_user)
put_text(f'你好，{current_user.nick}')
data = await input_group(
'大厅', inputs=[actions(name='cmd', buttons=['创建房间', '加入房间'])]
)
if data['cmd'] == '创建房间':
room_config = await input_group('房间设置', inputs=[
input(name='wolf_num', label='普通狼数', type=NUMBER, value='3'),
checkbox(name='god_wolf', label='特殊狼', inline=True, options=Role.as_god_wolf_options()),
input(name='citizen_num', label='普通村民数', type=NUMBER, value='4'),
checkbox(name='god_citizen', label='特殊村民', inline=True, options=Role.as_god_citizen_options()),
select(name='witch_rule', label='女巫解药规则', options=WitchRule.as_options()),
select(name='guard_rule', label='守卫规则', options=GuardRule.as_options()),
])
room = Room.alloc(room_config)
elif data['cmd'] == '加入房间':
room = Room.get(await input('房间号', type=TEXT, validate=Room.validate_room_join))
else:
raise NotImplementedError
put_scrollable(current_user.game_msg, height=200, keep_bottom=True)
current_user.game_msg.append(put_text(room.desc()))
room.add_player(current_user)
while True:
await asyncio.sleep(0.2)
# 非夜晚房主操作
host_ops = []
if current_user is room.get_host():
if not room.started:
host_ops = [
actions(name='host_op', buttons=['开始游戏'], help_text='你是房主')
]
elif room.stage == GameStage.Day and room.round > 0:
host_ops = [
actions(
name='host_vote_op',
buttons=[user.nick for user in room.list_alive_players()],
help_text='你是房主，本轮需要选择出局玩家'
)
]
# 玩家操作
user_ops = []
if room.started:
if room.stage == GameStage.WOLF and current_user.should_act():
user_ops = [
actions(
name='wolf_team_op',
buttons=add_cancel_button([user.nick for user in room.list_alive_players()]),
help_text='狼人阵营，请选择要击杀的对象。'
)
]
if room.stage == GameStage.DETECTIVE and current_user.should_act():
user_ops = [
actions(
name='detective_team_op',
buttons=[user.nick for user in room.list_alive_players()],
help_text='预言家，请选择要查验的对象。'
)
]
if room.stage == GameStage.WITCH and current_user.should_act():
if current_user.witch_has_heal():
current_user.send_msg(f'昨晚被杀的是 {room.list_pending_kill_players()}')
else:
current_user.send_msg('你已经没有解药了')
user_ops = [
radio(name='witch_mode', options=['解药', '毒药'], required=True, inline=True),
actions(
name='witch_team_op',
buttons=add_cancel_button([user.nick for user in room.list_alive_players()]),
help_text='女巫，请选择你的操作。'
)
]
if room.stage == GameStage.GUARD and current_user.should_act():
user_ops = [
actions(
name='guard_team_op',
buttons=add_cancel_button([user.nick for user in room.list_alive_players()]),
help_text='守卫，请选择你的操作。'
)
]
if room.stage == GameStage.HUNTER and current_user.should_act():
current_user.hunter_gun_status()
ops = host_ops + user_ops
if not ops:
continue
# UI
if host_ops + user_ops:
current_user.input_blocking = True
data = await input_group('操作', inputs=host_ops + user_ops, cancelable=True)
current_user.input_blocking = False
# Canceled
if data is None:
current_user.skip()
continue
# Host logic
if data.get('host_op') == '开始游戏':
await room.start_game()
if data.get('host_vote_op'):
await room.vote_kill(data.get('host_vote_op'))
# Wolf logic
if data.get('wolf_team_op'):
current_user.wolf_kill_player(nick=data.get('wolf_team_op'))
# Detective logic
if data.get('detective_team_op'):
current_user.detective_identify_player(nick=data.get('detective_team_op'))
# Witch logic
if data.get('witch_team_op'):
if data.get('witch_mode') == '解药':
current_user.witch_heal_player(nick=data.get('witch_team_op'))
elif data.get('witch_mode') == '毒药':
current_user.witch_kill_player(nick=data.get('witch_team_op'))
# Guard logic
if data.get('guard_team_op'):
current_user.guard_protect_player(nick=data.get('guard_team_op'))
if __name__ == '__main__':
logger.info(f"狼人杀服务器启动成功！可以通过在浏览器内输入 http://{get_interface_ip()} 来加入游戏")
start_server(main, debug=False, host='0.0.0.0', port=80, cdn=False)