294 lines
11 KiB
Python
294 lines
11 KiB
Python
"""
|
|
Creates a new SeleniumBase presentation with boilerplate code.
|
|
|
|
Usage:
|
|
seleniumbase mkpres [FILE.py] [LANG]
|
|
or sbase mkpres [FILE.py] [LANG]
|
|
|
|
Example:
|
|
sbase mkpres new_presentation.py --en
|
|
|
|
Language Options:
|
|
--en / --English | --zh / --Chinese
|
|
--nl / --Dutch | --fr / --French
|
|
--it / --Italian | --ja / --Japanese
|
|
--ko / --Korean | --pt / --Portuguese
|
|
--ru / --Russian | --es / --Spanish
|
|
|
|
Output:
|
|
Creates a new presentation with 3 example slides.
|
|
If the file already exists, an error is raised.
|
|
By default, the slides are written in English,
|
|
and use "serif" theme with "slide" transition.
|
|
The slides can be used as a basic boilerplate.
|
|
"""
|
|
import codecs
|
|
import colorama
|
|
import os
|
|
import sys
|
|
|
|
|
|
def invalid_run_command(msg=None):
|
|
exp = " ** mkpres **\n\n"
|
|
exp += " Usage:\n"
|
|
exp += " seleniumbase mkpres [FILE.py] [LANG]\n"
|
|
exp += " OR sbase mkpres [FILE.py] [LANG]\n"
|
|
exp += " Example:\n"
|
|
exp += " sbase mkpres new_presentation.py --en\n"
|
|
exp += " Language Options:\n"
|
|
exp += " --en / --English | --zh / --Chinese\n"
|
|
exp += " --nl / --Dutch | --fr / --French\n"
|
|
exp += " --it / --Italian | --ja / --Japanese\n"
|
|
exp += " --ko / --Korean | --pt / --Portuguese\n"
|
|
exp += " --ru / --Russian | --es / --Spanish\n"
|
|
exp += " Output:\n"
|
|
exp += " Creates a new presentation with 3 example slides.\n"
|
|
exp += " If the file already exists, an error is raised.\n"
|
|
exp += " By default, the slides are written in English,\n"
|
|
exp += ' and use "serif" theme with "slide" transition.\n'
|
|
exp += " The slides can be used as a basic boilerplate.\n"
|
|
if not msg:
|
|
raise Exception("INVALID RUN COMMAND!\n\n%s" % exp)
|
|
elif msg == "help":
|
|
print("\n%s" % exp)
|
|
sys.exit()
|
|
else:
|
|
raise Exception("INVALID RUN COMMAND!\n\n%s\n%s\n" % (exp, msg))
|
|
|
|
|
|
def main():
|
|
c1 = ""
|
|
c5 = ""
|
|
c7 = ""
|
|
cr = ""
|
|
if "linux" not in sys.platform:
|
|
c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
|
|
c5 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX
|
|
c7 = colorama.Fore.BLACK + colorama.Back.MAGENTA
|
|
cr = colorama.Style.RESET_ALL
|
|
|
|
help_me = False
|
|
error_msg = None
|
|
invalid_cmd = None
|
|
language = "English"
|
|
|
|
command_args = sys.argv[2:]
|
|
file_name = command_args[0]
|
|
if file_name == "-h" or file_name == "--help":
|
|
invalid_run_command("help")
|
|
elif not file_name.endswith(".py"):
|
|
error_msg = 'File name must end with ".py"!'
|
|
elif "*" in file_name or len(str(file_name)) < 4:
|
|
error_msg = "Invalid file name!"
|
|
elif file_name.startswith("-"):
|
|
error_msg = 'File name cannot start with "-"!'
|
|
elif "/" in str(file_name) or "\\" in str(file_name):
|
|
error_msg = "File must be created in the current directory!"
|
|
elif os.path.exists(os.getcwd() + "/" + file_name):
|
|
error_msg = 'File "%s" already exists in this directory!' % file_name
|
|
if error_msg:
|
|
error_msg = c5 + "ERROR: " + error_msg + cr
|
|
invalid_run_command(error_msg)
|
|
|
|
if len(command_args) >= 2:
|
|
options = command_args[1:]
|
|
for option in options:
|
|
option = option.lower()
|
|
if option == "-h" or option == "--help":
|
|
help_me = True
|
|
elif option == "--en" or option == "--english":
|
|
language = "English"
|
|
elif option == "--zh" or option == "--chinese":
|
|
language = "Chinese"
|
|
elif option == "--nl" or option == "--dutch":
|
|
language = "Dutch"
|
|
elif option == "--fr" or option == "--french":
|
|
language = "French"
|
|
elif option == "--it" or option == "--italian":
|
|
language = "Italian"
|
|
elif option == "--ja" or option == "--japanese":
|
|
language = "Japanese"
|
|
elif option == "--ko" or option == "--korean":
|
|
language = "Korean"
|
|
elif option == "--pt" or option == "--portuguese":
|
|
language = "Portuguese"
|
|
elif option == "--ru" or option == "--russian":
|
|
language = "Russian"
|
|
elif option == "--es" or option == "--spanish":
|
|
language = "Spanish"
|
|
else:
|
|
invalid_cmd = "\n===> INVALID OPTION: >> %s <<\n" % option
|
|
invalid_cmd = invalid_cmd.replace(">> ", ">>" + c5 + " ")
|
|
invalid_cmd = invalid_cmd.replace(" <<", " " + cr + "<<")
|
|
invalid_cmd = invalid_cmd.replace(">>", c7 + ">>" + cr)
|
|
invalid_cmd = invalid_cmd.replace("<<", c7 + "<<" + cr)
|
|
help_me = True
|
|
break
|
|
if help_me:
|
|
invalid_run_command(invalid_cmd)
|
|
|
|
dir_name = os.getcwd()
|
|
file_path = "%s/%s" % (dir_name, file_name)
|
|
html_name = file_name.replace(".py", ".html")
|
|
|
|
hello = "Hello"
|
|
press_right_arrow = "Press Right Arrow"
|
|
add_text = "Add Text"
|
|
goodbye = "Goodbye"
|
|
class_name = "MyTestClass"
|
|
|
|
if language == "Chinese":
|
|
hello = "你好"
|
|
press_right_arrow = "按向右箭头"
|
|
add_text = "添加文本"
|
|
goodbye = "再见"
|
|
class_name = "我的测试类"
|
|
elif language == "Dutch":
|
|
hello = "Hallo"
|
|
press_right_arrow = "Druk op pijl rechts"
|
|
add_text = "Tekst Toevoegen"
|
|
goodbye = "Dag"
|
|
class_name = "MijnTestklasse"
|
|
elif language == "French":
|
|
hello = "Bonjour"
|
|
press_right_arrow = "Appuyer sur flèche droite"
|
|
add_text = "Ajouter Texte"
|
|
goodbye = "Au revoir"
|
|
class_name = "MaClasseDeTest"
|
|
elif language == "Italian":
|
|
hello = "Ciao"
|
|
press_right_arrow = "Premere la freccia destra"
|
|
add_text = "Aggiungi Testo"
|
|
goodbye = "Addio"
|
|
class_name = "MiaClasseDiTest"
|
|
elif language == "Japanese":
|
|
hello = "こんにちは"
|
|
press_right_arrow = "右矢印を押します"
|
|
add_text = "テキストを追加"
|
|
goodbye = "さようなら"
|
|
class_name = "私のテストクラス"
|
|
elif language == "Korean":
|
|
hello = "여보세요"
|
|
press_right_arrow = "오른쪽 화살표를 누르십시오"
|
|
add_text = "텍스트를 추가"
|
|
goodbye = "안녕"
|
|
class_name = "테스트_클래스"
|
|
elif language == "Portuguese":
|
|
hello = "Olá"
|
|
press_right_arrow = "Pressione a seta direita"
|
|
add_text = "Adicionar Texto"
|
|
goodbye = "Tchau"
|
|
class_name = "MinhaClasseDeTeste"
|
|
elif language == "Russian":
|
|
hello = "Привет"
|
|
press_right_arrow = "Нажмите стрелку вправо"
|
|
add_text = "Добавить Текст"
|
|
goodbye = "До свидания"
|
|
class_name = "МойТестовыйКласс"
|
|
elif language == "Spanish":
|
|
hello = "Hola"
|
|
press_right_arrow = "Presione la flecha derecha"
|
|
add_text = "Agregar Texto"
|
|
goodbye = "Adiós"
|
|
class_name = "MiClaseDePrueba"
|
|
|
|
import_line = "from seleniumbase import BaseCase"
|
|
main_line = "BaseCase.main(__name__, __file__)"
|
|
parent_class = "BaseCase"
|
|
if language != "English":
|
|
from seleniumbase.translate.master_dict import MD_F
|
|
|
|
import_line = MD_F.get_import_line(language)
|
|
parent_class = MD_F.get_lang_parent_class(language)
|
|
class_line = "class %s(%s):" % (class_name, parent_class)
|
|
settings = 'theme="serif", transition="slide"'
|
|
img_src_1 = 'src="https://seleniumbase.github.io/cdn/gif/chart_pres.gif"'
|
|
img_src_2 = 'src="https://seleniumbase.github.io/cdn/img/sb_logo_10.png"'
|
|
hello_page = (
|
|
"\n '<h1>%s</h1><hr /><br />'"
|
|
"\n '<p>%s</p>'"
|
|
"" % (hello, press_right_arrow)
|
|
)
|
|
add_text_page = (
|
|
"\n '<h2><b>*</b> %s <b>*</b></h2>'"
|
|
"\n '<img %s>'"
|
|
"" % (add_text, img_src_1)
|
|
)
|
|
goodbye_page = (
|
|
"\n '<h2>%s</h2>'"
|
|
"\n '<img %s>'"
|
|
"" % (goodbye, img_src_2)
|
|
)
|
|
|
|
data = []
|
|
data.append("%s" % import_line)
|
|
data.append("%s" % main_line)
|
|
data.append("")
|
|
data.append("")
|
|
data.append("%s" % class_line)
|
|
data.append(" def test_presentation(self):")
|
|
data.append(" self.create_presentation(%s)" % settings)
|
|
data.append(" self.add_slide(%s)" % hello_page)
|
|
data.append(" self.add_slide(%s)" % add_text_page)
|
|
data.append(" self.add_slide(%s)" % goodbye_page)
|
|
data.append(' self.begin_presentation(filename="%s")' % html_name)
|
|
data.append("")
|
|
|
|
new_data = []
|
|
if language == "English":
|
|
new_data = data
|
|
else:
|
|
from seleniumbase.translate.master_dict import MD
|
|
from seleniumbase.translate.master_dict import MD_L_Codes
|
|
|
|
md = MD.md
|
|
lang_codes = MD_L_Codes.lang
|
|
nl_code = lang_codes[language]
|
|
dl_code = lang_codes["English"]
|
|
for line in data:
|
|
found_swap = False
|
|
replace_count = line.count("self.") # Total possible replacements
|
|
for key in md.keys():
|
|
original = "self." + md[key][dl_code] + "("
|
|
if original in line:
|
|
replacement = "self." + md[key][nl_code] + "("
|
|
new_line = line.replace(original, replacement)
|
|
found_swap = True
|
|
replace_count -= 1
|
|
if replace_count == 0:
|
|
break # Done making replacements
|
|
else:
|
|
# There might be another method to replace in the line.
|
|
# Example: self.assert_true("Name" in self.get_title())
|
|
line = new_line
|
|
continue
|
|
if main_line in line:
|
|
new_main = "%s.main(__name__, __file__)" % parent_class
|
|
new_line = line.replace(main_line, new_main)
|
|
found_swap = True
|
|
if found_swap:
|
|
if new_line.endswith(" # noqa"): # Remove flake8 skip
|
|
new_line = new_line[0 : -len(" # noqa")]
|
|
new_data.append(new_line)
|
|
continue
|
|
new_data.append(line)
|
|
data = new_data
|
|
file = codecs.open(file_path, "w+", "utf-8")
|
|
file.writelines("\r\n".join(data))
|
|
file.close()
|
|
if " " not in file_name:
|
|
os.system("sbase print %s -n" % file_name)
|
|
elif '"' not in file_name:
|
|
os.system('sbase print "%s" -n' % file_name)
|
|
else:
|
|
os.system("sbase print '%s' -n" % file_name)
|
|
success = (
|
|
"\n" + c1 + '* Presentation: "' + file_name + '" was created! *'
|
|
"" + cr + "\n"
|
|
)
|
|
print(success)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
invalid_run_command()
|