generate_page_set.py 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. #!/usr/bin/env python
  2. # Copyright (c) 2019 The Chromium Authors. All rights reserved.
  3. # Use of this source code is governed by a BSD-style license that can be
  4. # found in the LICENSE file.
  5. """Script that generates a page_set for the webpages_playback.py script."""
  6. import jinja2
  7. import os
  8. PAGE_SET_TEMPLATE = 'page_set_template'
  9. PAGE_SET_DIR = 'page_sets'
  10. def main():
  11. created_page_sets = []
  12. while True:
  13. user_agent = raw_input('user agent? (mobile/desktop/tablet): ')
  14. url_name = raw_input('URL name? (eg: google): ')
  15. url = raw_input('URL? (eg: http://www.google.com): ')
  16. comment = raw_input('Reason for adding the URL? (eg: go/skia-skps-3-2019): ')
  17. with open(PAGE_SET_TEMPLATE) as f:
  18. t = jinja2.Template(f.read())
  19. subs = {
  20. 'user_agent': user_agent,
  21. 'url_name': url_name,
  22. 'url': url,
  23. 'comment': comment,
  24. }
  25. page_set_name = 'skia_%s_%s.py' % (url_name, user_agent)
  26. page_set_path = os.path.join(PAGE_SET_DIR, page_set_name)
  27. with open(page_set_path, 'w') as f:
  28. f.write(t.render(**subs))
  29. created_page_sets.append(page_set_path)
  30. print '\nPage set has been created in %s\n\n' % page_set_path
  31. keep_going = raw_input('Do you have more page sets to create? (y/n)')
  32. if keep_going != 'y':
  33. break
  34. print '\n\nSummarizing all created page sets:'
  35. for page_set_path in created_page_sets:
  36. print '* %s' % page_set_path
  37. if __name__ == '__main__':
  38. main()