template.py 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748
  1. import asyncio
  2. import json
  3. import os
  4. import re
  5. import threading
  6. from functools import partial
  7. from os import path
  8. import time
  9. from percy import percySnapshot
  10. from selenium.webdriver import Chrome
  11. from selenium.webdriver.support.ui import Select
  12. from pywebio.input import *
  13. from pywebio.output import *
  14. from pywebio.session import *
  15. here_dir = path.dirname(path.abspath(__file__))
  16. def get_visible_form(browser):
  17. forms = browser.find_elements_by_css_selector('#input-cards > div')
  18. for f in forms:
  19. if f.is_displayed():
  20. return f
  21. def basic_output():
  22. set_env(title="PyWebIO Test")
  23. set_scope('top')
  24. put_markdown('### Basic')
  25. for i in range(3):
  26. put_text('text_%s' % i)
  27. put_text('测试空格:20空格:[%s]结束' % (' ' * 20))
  28. for i in range(3):
  29. put_text('inline_text_%s' % i, inline=True)
  30. put_markdown("""### put_markdown 测试
  31. `行内代码`
  32. 无序列表:
  33. - 北京
  34. - 上海
  35. - 天津
  36. 有序列表:
  37. 1. 北京
  38. 2. 上海
  39. 3. 天津
  40. [链接](./#)
  41. ~~删除线~~
  42. """, lstrip=True)
  43. put_link('链接', '#')
  44. put_text('<hr/>:')
  45. put_html("<hr/>")
  46. put_markdown('### Style')
  47. style(put_text('Red'), 'color:red')
  48. style([
  49. put_text('Red'),
  50. put_markdown('~~del~~')
  51. ], 'color:red')
  52. put_table([
  53. ['A', 'B'],
  54. ['C', style(put_text('Red'), 'color:red')],
  55. ])
  56. put_collapse('title', style([
  57. put_text('text'),
  58. put_markdown('~~del~~'),
  59. ], 'margin-left:20px'), open=True)
  60. put_markdown('### Table')
  61. put_table([
  62. ['Name', 'Gender', 'Address'],
  63. ['Wang', 'M', 'China'],
  64. ['Liu', 'W', 'America'],
  65. ])
  66. put_table([
  67. ['Wang', 'M', 'China'],
  68. ['Liu', 'W', 'America'],
  69. ], header=['Name', 'Gender', 'Address'])
  70. put_table([
  71. {"Course": "OS", "Score": "80"},
  72. {"Course": "DB", "Score": "93"},
  73. ], header=["Course", "Score"])
  74. put_table([
  75. {"Course": "OS", "Score": "80"},
  76. {"Course": "DB", "Score": "93"},
  77. ], header=[("课程", "Course"), ("得分", "Score")])
  78. img_data = open(path.join(here_dir, 'assets', 'img.png'), 'rb').read()
  79. put_table([
  80. ['Type', 'Content'],
  81. ['text', '<hr/>'],
  82. ['html', put_html('X<sup>2</sup>')],
  83. ['buttons', put_buttons(['A', 'B'], onclick=None, small=True)],
  84. ['markdown', put_markdown('`awesome PyWebIO!`\n - 1\n - 2\n - 3')],
  85. ['file', put_file('hello.text', b'')],
  86. ['image', put_image(img_data)],
  87. ['table', put_table([
  88. ['A', 'B'],
  89. [put_markdown('`C`'), put_markdown('`D`')]
  90. ])]
  91. ])
  92. put_markdown('### Code')
  93. with use_scope('scroll_basis'):
  94. put_code(json.dumps(dict(name='pywebio', author='wangweimin'), indent=4), 'json')
  95. put_text('move ⬆ code block to screen ... :')
  96. with use_scope('scroll_basis_btns'):
  97. put_buttons(buttons=[
  98. ('BOTTOM', Position.BOTTOM),
  99. ('TOP', Position.TOP),
  100. ('MIDDLE', Position.MIDDLE),
  101. ], onclick=lambda pos: scroll_to('scroll_basis', pos))
  102. def show_popup():
  103. popup('Popup title', [
  104. put_html('<h3>Popup Content</h3>'),
  105. 'html: <br/>',
  106. put_table([
  107. ['Type', 'Content'],
  108. ['html', put_html('X<sup>2</sup>')],
  109. ['text', '<hr/>'],
  110. ['buttons', put_buttons(['A', 'B'], onclick=...)],
  111. ['markdown', put_markdown('`Awesome PyWebIO!`')],
  112. ['file', put_file('hello.text', b'')],
  113. ['table', put_table([['A', 'B'], ['C', 'D']])]
  114. ]),
  115. put_buttons(['close_popup()'], onclick=lambda _: close_popup())
  116. ], size=PopupSize.NORMAL)
  117. with use_scope('popup_btn'):
  118. put_buttons([('popup()', '')], onclick=[show_popup])
  119. def edit_row(choice, row):
  120. put_text("You click %s button at row %s" % (choice, row), scope='table_cell_buttons')
  121. with use_scope('table_cell_buttons'):
  122. put_table([
  123. ['Idx', 'Actions'],
  124. ['1', put_buttons(['edit', 'delete'], onclick=partial(edit_row, row=1))],
  125. ['2', put_buttons(['edit', 'delete'], onclick=partial(edit_row, row=2))],
  126. ['3', put_buttons(['edit', 'delete'], onclick=partial(edit_row, row=3))],
  127. ])
  128. with use_scope('put_buttons'):
  129. put_buttons(['A', 'B', 'C'], onclick=partial(put_text, scope='put_buttons'))
  130. put_markdown('### Image')
  131. put_image(img_data)
  132. from PIL.Image import open as pil_open
  133. pil_img = pil_open(path.join(here_dir, 'assets', 'img.png'))
  134. put_image(pil_img, width="30px")
  135. put_image('https://cdn.jsdelivr.net/gh/wang0618/pywebio/test/assets/img.png', height="50px")
  136. put_file('hello_word.txt', b'hello word!')
  137. put_collapse('Collapse', [
  138. 'text',
  139. put_markdown('~~删除线~~'),
  140. put_table([
  141. ['商品', '价格'],
  142. ['苹果', '5.5'],
  143. ['香蕉', '7'],
  144. ])
  145. ], open=True)
  146. put_collapse('title', 'something', open=True)
  147. put_scrollable('scrollable\n' * 20, max_height=50)
  148. put_markdown('### Scope')
  149. with use_scope('scope1'):
  150. put_text("to be cleared")
  151. clear()
  152. put_text('A') # 输出内容: A
  153. put_text('B', position=0) # 输出内容: B A
  154. put_text('C', position=-2) # 输出内容: B C A
  155. with use_scope('scope2'):
  156. put_text('scope2')
  157. put_text('scope2')
  158. put_text('D', position=1) # 输出内容: B D C A
  159. put_text('before=top again', scope='top')
  160. with use_scope('to_remove'):
  161. put_text('to remove')
  162. remove('to_remove')
  163. put_markdown('### Info')
  164. session_info = get_info()
  165. from django.http import HttpRequest
  166. from flask import Request
  167. from tornado.httputil import HTTPServerRequest
  168. from aiohttp.web import BaseRequest
  169. request_type = {
  170. 'tornado': HTTPServerRequest,
  171. 'flask': Request,
  172. 'django': HttpRequest,
  173. 'aiohttp': BaseRequest,
  174. }
  175. request_ok = isinstance(session_info.request, request_type.get(session_info.backend))
  176. if not request_ok:
  177. print('Error: request check error: backend %s, request type %s, class %s' %
  178. (session_info.backend, type(session_info.request).__name__, session_info.request))
  179. put_markdown(rf"""### 会话信息
  180. ```
  181. * `user_agent`:
  182. * `is_mobile` (bool): {session_info.user_agent.is_mobile}
  183. * `is_tablet` (bool): {session_info.user_agent.is_tablet}
  184. * `is_pc` (bool): {session_info.user_agent.is_pc}
  185. * `is_touch_capable` (bool): {session_info.user_agent.is_touch_capable}
  186. * `browser.family` (str): {session_info.user_agent.browser.family}
  187. * `os.family` (str): {session_info.user_agent.os.family}
  188. * `os.version` (tuple): {session_info.user_agent.os.version}
  189. * `os.version_string` (str): {session_info.user_agent.os.version_string}
  190. * `device.family` (str): {session_info.user_agent.device.family}
  191. * `device.brand` (str): {session_info.user_agent.device.brand}
  192. * `device.model` (str): {session_info.user_agent.device.model}
  193. * `user_language` (str): {session_info.user_language}
  194. * `server_host` (str): {session_info.server_host}
  195. * `origin` (str): {session_info.origin or 'http://' + session_info.server_host}
  196. * `user_ip` (str): {session_info.user_ip}
  197. * `request type check` (str): {request_ok}
  198. ```
  199. """, strip_indent=4)
  200. put_markdown('### Layout')
  201. put_row([
  202. put_column([
  203. put_code('A'),
  204. put_row([
  205. put_code('B1'), None,
  206. put_code('B2'), None,
  207. put_code('B3'),
  208. ]),
  209. put_code('C'),
  210. ]), None,
  211. put_code('python'), None,
  212. style(put_code('python\n' * 20), 'max-height:200px;'),
  213. ])
  214. put_grid([
  215. [style(put_code('[%s,%s]' % (x, y)), 'margin-right:10px;') for y in range(4)]
  216. for x in range(5)
  217. ], direction='column')
  218. put_row([style(put_code(i), 'margin-right:10px;') for i in range(4)], 'repeat(auto-fill, 25%)')
  219. put_markdown('### Span')
  220. cell = lambda text: style(put_code(text), 'margin-right:10px;')
  221. put_grid([
  222. [span(cell('A'), col=2), None],
  223. [span(cell('C'), row=2, col=2), span(cell('D'), row=2)],
  224. [],
  225. ], cell_width='1fr', cell_height='1fr')
  226. put_table([
  227. ['C'],
  228. [span('E', col=2)],
  229. ], header=[span('A', row=2), 'B'])
  230. put_processbar('processbar', 0.3)
  231. set_processbar('processbar', 0.6)
  232. put_loading()
  233. # output
  234. hobby = output(put_text('Coding'))
  235. put_table([
  236. ['Name', 'Hobbies'],
  237. ['Wang', hobby]
  238. ])
  239. hobby.reset(put_text('Movie'))
  240. hobby.append(put_text('Music'), put_text('Drama'))
  241. hobby.insert(0, put_markdown('**Coding**'))
  242. def background_output():
  243. put_text("Background output")
  244. set_scope('background')
  245. def background():
  246. for i in range(20):
  247. put_text('%s ' % i, inline=True, scope='background')
  248. t = threading.Thread(target=background)
  249. register_thread(t)
  250. t.start()
  251. async def coro_background_output():
  252. put_text("Background output")
  253. set_scope('background')
  254. async def background():
  255. for i in range(20):
  256. put_text('%s ' % i, inline=True, scope='background')
  257. return run_async(background())
  258. def test_output(browser: Chrome, enable_percy=False):
  259. """测试输出::
  260. run template.basic_output()
  261. template.background_output() # 或者 await template.coro_background_output()
  262. hold()
  263. """
  264. time.sleep(1) # 等待输出完毕
  265. # get focus
  266. browser.find_element_by_tag_name('body').click()
  267. time.sleep(1)
  268. browser.execute_script('$("html, body").scrollTop( $(document).height()+100);')
  269. time.sleep(0.5)
  270. enable_percy and percySnapshot(browser=browser, name='begin output')
  271. tab_btns = browser.find_elements_by_css_selector('#pywebio-scope-table_cell_buttons button')
  272. for btn in tab_btns:
  273. time.sleep(0.5)
  274. browser.execute_script("arguments[0].click();", btn)
  275. btns = browser.find_elements_by_css_selector('#pywebio-scope-put_buttons button')
  276. for btn in btns:
  277. time.sleep(0.5)
  278. browser.execute_script("arguments[0].click();", btn)
  279. # 滚动窗口
  280. btns = browser.find_elements_by_css_selector('#pywebio-scope-scroll_basis_btns button')
  281. for btn in btns:
  282. time.sleep(1)
  283. browser.execute_script("arguments[0].click();", btn)
  284. time.sleep(1)
  285. browser.execute_script('$("html, body").scrollTop( $(document).height()+100);')
  286. time.sleep(0.5)
  287. enable_percy and percySnapshot(browser=browser, name='basic output')
  288. # popup
  289. btn = browser.find_element_by_css_selector('#pywebio-scope-popup_btn button')
  290. browser.execute_script("arguments[0].click();", btn)
  291. time.sleep(1)
  292. enable_percy and percySnapshot(browser=browser, name='popup')
  293. browser.execute_script("$('.modal').modal('hide');")
  294. def basic_input():
  295. js_res = yield eval_js('''(function(){
  296. for(var i=0;i<=limit;i++)
  297. a += i;
  298. return a;
  299. })()''', a=0, limit=100)
  300. assert js_res == 5050
  301. age = yield input("How old are you?", type=NUMBER)
  302. put_markdown(f'`{repr(age)}`')
  303. password = yield input("Input password", type=PASSWORD)
  304. put_markdown(f'`{repr(password)}`')
  305. # 下拉选择框
  306. gift = yield select('Which gift you want?', ['keyboard', 'ipad'])
  307. put_markdown(f'`{repr(gift)}`')
  308. # CheckBox
  309. agree = yield checkbox("用户协议", options=['I agree to terms and conditions'])
  310. put_markdown(f'`{repr(agree)}`')
  311. # Text Area
  312. text = yield textarea('Text Area', rows=3, placeholder='Some text')
  313. put_markdown(f'`{repr(text)}`')
  314. # 文件上传
  315. img = yield file_upload("Select a image:", accept="image/*", max_size=10 ** 7)
  316. put_image(img['content'], title=img['filename'])
  317. # 输入参数
  318. res = yield input('This is label', type=TEXT, placeholder='This is placeholder,required=True',
  319. help_text='This is help text', required=True)
  320. put_markdown(f'`{repr(res)}`')
  321. # 取消表单
  322. res = yield input_group('cancel test', [input(name='cancel')], cancelable=True)
  323. put_markdown(f'`{repr(res)}`')
  324. # 校验函数
  325. def check_age(p): # 检验函数校验通过时返回None,否则返回错误消息
  326. if p < 10:
  327. return 'Too young!!'
  328. if p > 60:
  329. return 'Too old!!'
  330. age = yield input("How old are you?", type=NUMBER, validate=check_age, help_text='age in [10, 60]')
  331. put_markdown(f'`{repr(age)}`')
  332. # Codemirror
  333. code = yield textarea('Code Edit', code={
  334. 'mode': "python", # 编辑区代码语言
  335. 'theme': 'darcula', # 编辑区darcula主题, Visit https://codemirror.net/demo/theme.html#cobalt to get more themes
  336. }, value='import something\n# Write your python code')
  337. put_markdown(f'`{repr(code)}`')
  338. # 输入组 cancelable
  339. info = yield input_group("Cancelable", [
  340. input('Input your name', name='name'),
  341. input('Input your age', name='age', type=NUMBER, validate=check_age, help_text='age in [10, 60]')
  342. ], cancelable=True)
  343. put_markdown(f'`{repr(info)}`')
  344. # input action
  345. def set_now_ts(set_value):
  346. set_value('set from action')
  347. val = yield input('Input action', action=('Set value', set_now_ts))
  348. assert val == 'set from action'
  349. def check_form(data): # 检验函数校验通过时返回None,否则返回 (input name,错误消息)
  350. if len(data['password']) > 6:
  351. return ('password', 'password太长!')
  352. check_item_data = []
  353. def check_item(data):
  354. check_item_data.append(repr(data))
  355. info = yield input_group('Input group', [
  356. input('Text', type=TEXT, datalist=['data-%s' % i for i in range(10)], name='text',
  357. required=True, help_text='required=True', validate=check_item),
  358. input('Number', type=NUMBER, value="42", name='number', validate=check_item),
  359. input('Float', type=FLOAT, name='float', validate=check_item),
  360. input('Password', type=PASSWORD, name='password', validate=check_item),
  361. textarea('Textarea', rows=3, maxlength=20, name='textarea',
  362. help_text='rows=3, maxlength=20', validate=check_item),
  363. textarea('Code', name='code', code={
  364. 'lineNumbers': False,
  365. 'indentUnit': 2,
  366. }, value='import something\n# Write your python code', validate=check_item),
  367. select('select-multiple', [
  368. {'label': '标签0,selected', 'value': '0', 'selected': True},
  369. {'label': '标签1,disabled', 'value': '1', 'disabled': True},
  370. ('标签2,selected', '2', True),
  371. ('标签3', '3'),
  372. ('标签4,disabled', '4', False, True),
  373. '标签5,selected',
  374. ], name='select-multiple', multiple=True, value=['标签5,selected'], required=True,
  375. help_text='required至少选择一项', validate=check_item),
  376. select('select', [
  377. {'label': '标签0', 'value': '0', 'selected': False},
  378. {'label': '标签1,disabled', 'value': '1', 'disabled': True},
  379. ('标签2', '2', False),
  380. ('标签3', '3'),
  381. ('标签4,disabled', '4', False, True),
  382. '标签5,selected',
  383. ], name='select', value=['标签5,selected'], validate=check_item),
  384. checkbox('checkbox-inline', [
  385. {'label': '标签0,selected', 'value': '0', 'selected': False},
  386. {'label': '标签1,disabled', 'value': '1', 'disabled': True},
  387. ('标签2,selected', '2', True),
  388. ('标签3', '3'),
  389. ('标签4,disabled', '4', False, True),
  390. '标签5,selected',
  391. ], inline=True, name='checkbox-inline', value=['标签5,selected', '标签0', '标签0,selected'], validate=check_item),
  392. checkbox('checkbox', [
  393. {'label': '标签0,selected', 'value': '0', 'selected': True},
  394. {'label': '标签1,disabled', 'value': '1', 'disabled': True},
  395. ('标签2,selected', '2', True),
  396. ('标签3', '3'),
  397. ('标签4,disabled', '4', False, True),
  398. '标签5',
  399. ], name='checkbox', validate=check_item),
  400. radio('radio-inline', [
  401. {'label': '标签0', 'value': '0', 'selected': False},
  402. {'label': '标签1,disabled', 'value': '1', 'disabled': True},
  403. ('标签2', '2', False),
  404. ('标签3', '3'),
  405. ('标签4,disabled', '4', False, True),
  406. '标签5,selected',
  407. ], inline=True, name='radio-inline', value='标签5,selected', validate=check_item),
  408. radio('radio', [
  409. {'label': '标签0', 'value': '0', 'selected': False},
  410. {'label': '标签1,disabled', 'value': '1', 'disabled': True},
  411. ('标签2', '2', False),
  412. ('标签3', '3'),
  413. ('标签4,disabled', '4', False, True),
  414. '标签5,selected',
  415. ], inline=False, name='radio', value='标签5,selected', validate=check_item),
  416. file_upload('file_upload', name='file_upload', max_size='10m'),
  417. actions('actions', [
  418. {'label': '提交', 'value': 'submit'},
  419. ('提交2', 'submit2'),
  420. '提交3',
  421. {'label': 'disabled', 'disabled': True},
  422. ('重置', 'reset', 'reset'),
  423. {'label': '取消', 'type': 'cancel'},
  424. ], name='actions', help_text='actions'),
  425. ], validate=check_form)
  426. put_text('`validate()` log:')
  427. put_code(json.dumps(sorted(list(set(check_item_data))), indent=4, ensure_ascii=False), 'json')
  428. put_text('Form result:')
  429. if info:
  430. put_code(json.dumps([repr(i) for i in sorted(info.items())], indent=4, ensure_ascii=False), 'json')
  431. # yield actions(['Continue'])
  432. def background_input():
  433. def background():
  434. time.sleep(1)
  435. res = input('background')
  436. put_markdown(f'`background: {repr(res)}`')
  437. t = threading.Thread(target=background)
  438. register_thread(t)
  439. t.start()
  440. res = input('front')
  441. put_markdown(f'`front: {repr(res)}`')
  442. async def coro_background_input():
  443. async def background():
  444. await asyncio.sleep(1)
  445. res = await input('background')
  446. put_markdown(f'`background: {repr(res)}`')
  447. run_async(background())
  448. res = await input('front')
  449. put_markdown(f'`front: {repr(res)}`')
  450. async def flask_coro_background_input():
  451. async def background():
  452. await run_asyncio_coroutine(asyncio.sleep(1))
  453. res = await input('background')
  454. put_markdown(f'`background: {repr(res)}`')
  455. run_async(background())
  456. res = await input('front')
  457. put_markdown(f'`front: {repr(res)}`')
  458. def test_input(browser: Chrome, enable_percy=False):
  459. """测试输入::
  460. run template.basic_input()
  461. actions(['Continue'])
  462. template.background_input() # 或者 await template.coro_background_input() / flask_coro_background_input
  463. """
  464. browser.find_element_by_css_selector('input').send_keys("22")
  465. browser.find_element_by_tag_name('form').submit()
  466. time.sleep(0.5)
  467. browser.find_element_by_css_selector('input').send_keys("secret")
  468. browser.find_element_by_tag_name('form').submit()
  469. time.sleep(0.5)
  470. browser.find_element_by_tag_name('form').submit()
  471. # checkbox
  472. time.sleep(0.5)
  473. browser.execute_script("arguments[0].click();", browser.find_element_by_css_selector('input'))
  474. browser.find_element_by_tag_name('form').submit()
  475. # Text Area
  476. time.sleep(0.5)
  477. browser.find_element_by_css_selector('textarea').send_keys(" ".join(str(i) for i in range(20)))
  478. browser.find_element_by_tag_name('form').submit()
  479. # file
  480. time.sleep(0.5)
  481. img_path = path.join(here_dir, 'assets', 'img.png')
  482. browser.find_element_by_css_selector('input').send_keys(img_path)
  483. browser.find_element_by_tag_name('form').submit()
  484. # text
  485. time.sleep(0.5)
  486. browser.find_element_by_css_selector('input').send_keys("text")
  487. browser.find_element_by_tag_name('form').submit()
  488. # 表单取消
  489. time.sleep(0.5)
  490. browser.execute_script("arguments[0].click();", browser.find_element_by_css_selector('.pywebio_cancel_btn'))
  491. # valid func, age in [10, 60]
  492. time.sleep(0.5)
  493. browser.find_element_by_css_selector('input').send_keys("1")
  494. browser.find_element_by_tag_name('form').submit()
  495. time.sleep(0.5)
  496. browser.find_element_by_css_selector('input').clear()
  497. browser.find_element_by_css_selector('input').send_keys("90")
  498. browser.find_element_by_tag_name('form').submit()
  499. time.sleep(0.5)
  500. browser.find_element_by_css_selector('input').clear()
  501. browser.find_element_by_css_selector('input').send_keys("23")
  502. browser.find_element_by_tag_name('form').submit()
  503. # code
  504. time.sleep(0.5)
  505. # browser.find_element_by_css_selector('textarea').send_keys(" ".join(str(i) for i in range(20)))
  506. browser.find_element_by_tag_name('form').submit()
  507. # Cancelable from group
  508. time.sleep(0.5)
  509. browser.find_element_by_name('name').send_keys("name")
  510. time.sleep(1)
  511. browser.find_element_by_name('age').send_keys("90")
  512. browser.find_element_by_tag_name('form').submit()
  513. browser.execute_script('$("html, body").scrollTop( $(document).height()+100);')
  514. time.sleep(0.5)
  515. enable_percy and percySnapshot(browser=browser, name='input group invalid')
  516. time.sleep(0.5)
  517. browser.find_element_by_name('age').clear()
  518. browser.find_element_by_name('age').send_keys("23")
  519. browser.find_element_by_tag_name('form').submit()
  520. # callback actions
  521. time.sleep(0.5)
  522. browser.execute_script("arguments[0].click();", browser.find_element_by_css_selector('form button[type="button"]'))
  523. time.sleep(0.4)
  524. # input action
  525. time.sleep(0.5)
  526. browser.execute_script("arguments[0].click();", browser.find_element_by_css_selector('form button[type="button"]'))
  527. time.sleep(0.4)
  528. browser.find_element_by_tag_name('form').submit()
  529. # Input group
  530. time.sleep(0.5)
  531. browser.execute_script('$("html, body").scrollTop( $(document).height()+100);')
  532. time.sleep(0.5)
  533. enable_percy and percySnapshot(browser=browser, name='input group all')
  534. browser.find_element_by_name('text').send_keys("name")
  535. browser.find_element_by_name('number').send_keys("20")
  536. browser.find_element_by_name('float').send_keys("3.1415")
  537. browser.find_element_by_name('password').send_keys("password")
  538. browser.find_element_by_name('textarea').send_keys(" ".join(str(i) for i in range(20)))
  539. # browser.find_element_by_css_selector('[name="code"]').send_keys(" ".join(str(i) for i in range(10)))
  540. Select(browser.find_element_by_name('select-multiple')).select_by_index(0)
  541. # browser. find_element_by_css_selector('[name="select"]'). send_keys("name")
  542. # browser. find_element_by_css_selector('[name="checkbox-inline"]'). send_keys("name")
  543. # browser. find_element_by_css_selector('[name="checkbox"]'). send_keys("name")
  544. # browser. find_element_by_css_selector('[name="radio-inline"]'). send_keys("name")
  545. # browser. find_element_by_css_selector('[name="radio"]'). send_keys("name")
  546. browser.find_element_by_name('file_upload').send_keys(path.join(here_dir, 'assets', 'helloworld.txt'))
  547. browser.execute_script("$('form button').eq(1).click()")
  548. time.sleep(1)
  549. browser.execute_script('$("html, body").scrollTop( $(document).height()+100);')
  550. time.sleep(0.5)
  551. enable_percy and percySnapshot(browser=browser, name='input group all invalid')
  552. browser.find_element_by_name('password').clear()
  553. browser.find_element_by_name('password').send_keys("123")
  554. browser.execute_script("$('form button').eq(1).click()")
  555. time.sleep(1)
  556. browser.execute_script('$("html, body").scrollTop( $(document).height()+100);')
  557. time.sleep(1)
  558. enable_percy and percySnapshot(browser=browser, name='input group all submit')
  559. browser.find_element_by_css_selector('form').submit()
  560. # background
  561. time.sleep(3)
  562. get_visible_form(browser).find_element_by_css_selector('input').send_keys("background")
  563. get_visible_form(browser).find_element_by_tag_name('form').submit()
  564. # front
  565. time.sleep(0.5)
  566. get_visible_form(browser).find_element_by_css_selector('input').send_keys("front")
  567. get_visible_form(browser).find_element_by_tag_name('form').submit()
  568. def set_defer_call():
  569. def deferred_1():
  570. open('test_defer.tmp', 'w').write('deferred_1')
  571. def deferred_2():
  572. open('test_defer.tmp', 'a').write('deferred_2')
  573. defer_call(deferred_1)
  574. defer_call(deferred_2)
  575. def test_defer_call():
  576. output = open('test_defer.tmp').read()
  577. assert "deferred_1" in output
  578. assert "deferred_2" in output
  579. os.remove('test_defer.tmp')
  580. def save_output(browser: Chrome, filename=None, process_func=None):
  581. """获取输出区html源码,并去除随机元素,供之后diff比较
  582. :param browser:
  583. :param filename: 保存文件名, 为 None 时,不保存为文件
  584. :param process_func: 自定义数据处理函数
  585. :return: 处理前后的html文本
  586. """
  587. raw_html = browser.find_element_by_id('markdown-body').get_attribute('innerHTML')
  588. html = re.sub(r'"pywebio-scope-.*?"', '', raw_html)
  589. html = re.sub(r"WebIO.pushData\(.*?\)", '', html)
  590. html = re.sub(r"</(.*?)>", r'</\g<1>>\n', html) # 进行断行方便后续的diff判断
  591. html = html.replace('"opacity: 1;"', '').replace(' open=""', '') # so wired
  592. if process_func:
  593. html = process_func(html)
  594. if filename:
  595. open(path.join(here_dir, 'output', filename), 'w').write(html)
  596. return raw_html, html