def test_is_area_correct(self):
self.assertTrue(StringUtils.is_area_correct('(0,0,0,0)'))
self.assertTrue(StringUtils.is_area_correct('(0, 0, 0, 0)'))
self.assertTrue(StringUtils.is_area_correct('( 0, 0, 0, 0 )'))
self.assertTrue(StringUtils.is_area_correct('(100, 100, 100, 100)'))
self.assertFalse(StringUtils.is_area_correct('(0, 0, 0)'))
self.assertFalse(StringUtils.is_area_correct('(0, 0, 0, d)'))
self.assertFalse(StringUtils.is_area_correct('0, 0, 0, 0'))
self.assertFalse(StringUtils.is_area_correct('(0, 0, 0, 0, 0)'))
self.assertFalse(StringUtils.is_area_correct('[0, 0, 0, 0]'))
def test_is_area_correct(self):
self.assertTrue(StringUtils.is_area_correct('(0,0,0,0)'))
self.assertTrue(StringUtils.is_area_correct('(0, 0, 0, 0)'))
self.assertTrue(
StringUtils.is_area_correct('( 0, 0, 0, 0 )'))
self.assertTrue(StringUtils.is_area_correct('(100, 100, 100, 100)'))
self.assertFalse(StringUtils.is_area_correct('(0, 0, 0)'))
self.assertFalse(StringUtils.is_area_correct('(0, 0, 0, d)'))
self.assertFalse(StringUtils.is_area_correct('0, 0, 0, 0'))
self.assertFalse(StringUtils.is_area_correct('(0, 0, 0, 0, 0)'))
self.assertFalse(StringUtils.is_area_correct('[0, 0, 0, 0]'))
def generate(self, evt):
if self.__is_gen_data_correct():
folder = self.__get_root_folder()
if RootFolder.PO_FOLDER in os.listdir(folder):
folder = os.path.join(folder, RootFolder.PO_FOLDER)
class_name = self.txt_class_name.GetValue()
file_path = os.path.join(folder,
get_py_file_name_from_class_name(class_name))
area_as_text = self.txt_selected_area.GetValue()
url = self.main_frame.get_url()
if os.path.exists(file_path):
show_dialog_path_does_exist(self, file_path)
elif not StringUtils.is_class_name_correct(class_name):
show_dialog_bad_name(self, class_name, 'Header', 'ContextMenu')
elif not StringUtils.is_area_correct(area_as_text):
show_dialog(self, u'Bad selected area: %s' % area_as_text,
u'Bad selected area')
elif not StringUtils.is_url_correct(url):
show_dialog(self, u'Bad url: %s' % url, u'Bad url')
else:
dialog = DialogWithText(self, 'Generating page object class...')
handler = WxTextCtrlHandler(dialog.txt_ctrl)
logger = Logger(log_to_console=False, handler=handler)
dialog.Show()
area = eval(area_as_text)
generator = PageObjectGenerator(self.main_frame.get_browser(),
logger)
folder_path = self.main_frame.get_tmp_dir()
def generate():
dialog.btn_ok.Disable()
po_class = generator.get_po_class_for_url(url,
class_name,
folder_path,
area)
po_class.save(folder)
logger.info(u"Saving class '%s'..." % po_class.name)
logger.info(u'Saved file: %s' % po_class.file_path)
logger.info(u'Saved file: %s' % po_class.img_path)
logger.info(u'DONE')
dialog.btn_ok.Enable()
thread = Thread(target=generate)
thread.setDaemon(True)
thread.start()
def generate(self, evt):
if self.__is_gen_data_correct():
folder = self.__get_root_folder()
if RootFolder.PO_FOLDER in os.listdir(folder):
folder = os.path.join(folder, RootFolder.PO_FOLDER)
class_name = self.txt_class_name.GetValue()
file_path = os.path.join(
folder, get_py_file_name_from_class_name(class_name))
area_as_text = self.txt_selected_area.GetValue()
url = self.main_frame.get_url()
if os.path.exists(file_path):
show_dialog_path_does_exist(self, file_path)
elif not StringUtils.is_class_name_correct(class_name):
show_dialog_bad_name(self, class_name, 'Header', 'ContextMenu')
elif not StringUtils.is_area_correct(area_as_text):
show_dialog(self, u'Bad selected area: %s' % area_as_text,
u'Bad selected area')
elif not StringUtils.is_url_correct(url):
show_dialog(self, u'Bad url: %s' % url, u'Bad url')
else:
dialog = DialogWithText(self,
'Generating page object class...')
handler = WxTextCtrlHandler(dialog.txt_ctrl)
logger = Logger(log_to_console=False, handler=handler)
dialog.Show()
area = eval(area_as_text)
generator = PageObjectGenerator(self.main_frame.get_browser(),
logger)
folder_path = self.main_frame.get_tmp_dir()
def generate():
dialog.btn_ok.Disable()
po_class = generator.get_po_class_for_url(
url, class_name, folder_path, area)
po_class.save(folder)
logger.info(u"Saving class '%s'..." % po_class.name)
logger.info(u'Saved file: %s' % po_class.file_path)
logger.info(u'Saved file: %s' % po_class.img_path)
logger.info(u'DONE')
dialog.btn_ok.Enable()
thread = Thread(target=generate)
thread.setDaemon(True)
thread.start()
