import unittest
import html
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import selenium.webdriver.support.ui as ui
from selenium.common.exceptions import TimeoutException
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import StaleElementReferenceException
################## הגדרות #############
#אל תשאל אותי מה זה
penelope = True
#שם משתמש
un = "gondolini"
#סיסמה
ps = "123123"
#שמות דפים
pagesNames = ['דף ראשון','דף שני','דף שלישי','דף רביעי','דף חמישי']
#תוכן דפים
pagesContent = ['תוכן ראשון','תוכן שני','תוכן שלישי','תוכן רביעי','תוכן חמישי']

epi = []
epi.append({'start' : 1, 'end' : 39, 'img' : 'http://www.seret.club/wp-content/uploads/2016/09/%D7%AA%D7%90%D7%92%D7%93_33_%D7%9C%D7%A6%D7%A4%D7%99%D7%99%D7%94_%D7%99%D7%A9%D7%99%D7%A8%D7%94.jpg', 'prog' : 'תאג"ד'})
epi.append({'start' : 1, 'end' : 6, 'img' : 'http://10tv.online/wp-content/uploads/2016/10/%D7%9B%D7%97%D7%95%D7%9C%D7%99%D7%9D-%D7%9C%D7%A6%D7%A4%D7%99%D7%99%D7%94-%D7%99%D7%A9%D7%99%D7%A8%D7%94.jpg', 'prog' : 'כחולים'})
epi.append({'start' : 1, 'end' : 24, 'img' : 'http://10tv.online/wp-content/uploads/2016/10/master.jpg', 'prog' : 'מאסטר שף עונה 6'})
epi.append({'start' : 1, 'end' : 50, 'img' : 'http://10tv.online/wp-content/uploads/2016/10/13895128_1166070643434815_8944144051395233481_n.png', 'prog' : 'מטומטמת'})
epi.append({'start' : 1, 'end' : 10, 'img' : 'http://watchgalis.com/wp-content/uploads/2016/09/%D7%A2%D7%A7%D7%A8%D7%95%D7%9F-%D7%94%D7%94%D7%97%D7%9C%D7%A4%D7%94-%D7%A4%D7%A8%D7%A7-1-%D7%9C%D7%A6%D7%A4%D7%99%D7%99%D7%94-%D7%99%D7%A9%D7%99%D7%A8%D7%94.png', 'prog' : 'עקרון ההחלפה'})
epi.append({'start' : 1, 'end' : 50, 'img' : 'http://1.bp.blogspot.com/-Va9wDWKuNss/VENoUpC1plI/AAAAAAAAClU/eiBfE9qaFPU/s1600/1937480_831773500174430_1432024979625445154_n.png', 'prog' : 'שכונה עונה 2'})
epi.append({'start' : 1, 'end' : 20, 'img' : 'http://10tv.online/wp-content/uploads/2016/10/maslul.jpg', 'prog' : 'מחשב מסלול מחדש'})
epi.append({'start' : 1, 'end' : 20, 'img' : 'http://10tv.online/wp-content/uploads/2016/10/קרדיט-צילום-עידו-לביא20163030_3878750_0..jpg', 'prog' : 'מסובך'})
def getCon(name,mogi,img):
    stri = '<img src="'+img+'" height="200px" width="500px"/> <h1 style="text-align:center"><a href="http://10tv.online/שכונה-עונה-2-פרק-1-לצפייה-ישירה"> שכונה עונה 2 פרק 1 לצפייה ישירה </a> </h1> <p> הרבה זמן שלא דיברנו על <a href="http://10tv.online/שכונה-עונה-2-פרק-1-לצפייה-ישירה"> שכונה עונה 2 פרק 1 לצפייה ישירה </a>, אולי זה מאחר והעלילה שלו כזו טובה? אולי מאחר והמבקרים לא מפסיקים לשבח את הסדרה המדהימה הזו, וטוענים כי הם הדבר הכי טוב שהם ראו אי פעם בחיים שלהם. אבל אם אתם שואלים אותי, המבקר הכי טוב בעולם, האלמנטים בעלילה של שכונה עונה 2 פרק 1 לצפייה ישירה הם מה שעושה אותה לסדרה כזו חזקה, עלילתית ומעניינת, ושלא נדבר על זה שהיא מאתגרת. </p> <h3 style="color:red;text-align:center"><a href="http://10tv.online/שכונה-עונה-2-פרק-1-לצפייה-ישירה"> לצפייה בפרק לחצו כאן </a> </h3>'
    mog = name.replace(' ','-')
    stri = stri.replace("שכונה-עונה-2-פרק-1-לצפייה-ישירה",mog)
    stri = stri.replace("שכונה עונה 2 פרק 1 לצפייה ישירה",name)
    stri = stri + getTags(mogi)
    return (stri)
def getTags(name):
    return ("<br>" + name + "<br>" + name + " לצפייה ישירה<br>" + name + " חינם<br>" + name + " הפרק המלא<br>" + name + " לצפייה חינם")
def tags(name):
    return (name + " הפרק המלא," + name + " לצפייה ישירה חינם," + name + " לצפייה ישירה," + name + " לצפייה ישירה הפרק המלא," + name)

########### FUNCTIONS ############
def from_to(st,fr,to):
    pos = st.find(fr)
    st = st[pos:]
    pos = st.find(to)
    st = st[:pos]
    return (st)
def rand_key():
    import random
    m = random.randint(0,999999)
    return ("chapter_view_"+str(m))
def rand_login():
    import random
    m = random.randint(0,999999)
    return ("mombo"+str(m))
def add_pages(driver):
    driver.find_element_by_id("a_AddPages").click()
    driver.find_element_by_id("PageAddName").send_keys(rand_key())
    driver.find_element_by_id("BtnPageAdd").click()
def find_a(st):
    pos = st.find('"')
    return ('http://www.CMSAdmin.co.il/'+st[:pos])
def wait(g):
    for i in range(1,g):
        print("W A I T . . . ")
######### INPUTS ##########
        """
prog = input("Program: ")
start = input("Start: ")
end = input("End: ")
"""
doLogin = input("Login? ")
openBlogs = input("Open Blogs? ")
if(openBlogs == "1"):
    blogsNum = input("How Many Blogs:")
addPages = input("Add Pages To Blogs? ")
openUsers = input("Register Users? ")
if(openUsers == "1"):
    usersNum = input("How Many Users:")


driver = webdriver.Firefox()
#התחברות
if(doLogin == "1"):
    driver.get("http://web.2all.co.il")
    driver.execute_script("document.getElementsByClassName('dropdown-menu')[2].style.display = 'block';")
    username = driver.find_element_by_id("txtUsername")
    username.send_keys(un)
    password = driver.find_element_by_id("txtPassword")
    password.send_keys(ps)
    driver.execute_script("javascript:funcLogin(0,1);")
#פתיחת בלוגים
if(openBlogs == "1"):
    for i in range(int(blogsNum)):
        driver.implicitly_wait(10)
        driver.find_element_by_partial_link_text("+ בנה אתר נוסף").click()
        for handle in driver.window_handles:
            driver.switch_to_window(handle)

        """ בניית בלוג """
        driver.execute_script("SubTemplateSet(409889)")
        driver.find_element_by_id("TxtSBDomain").send_keys(rand_key())
        driver.find_element_by_id("Btn_Step3_Save").click()
        driver.execute_script("window.close()")
        for handle in driver.window_handles:
            driver.switch_to_window(handle)

#הוספת דפים
if(addPages == "1"):
    print('motek')
    driver.implicitly_wait(10)
    driver.find_element_by_class_name("toggle").click()
    html1 = driver.page_source
    div = from_to(html1,'id="tblMoreWebsites"','</table>')
    div = div.split('http://www.CMSAdmin.co.il/')
    arr = []
    for i in div:
            if(i != div[0]):
                arr.append(html.unescape(find_a(i)))
    #הוספת דפים לכל בלוג
    for i in arr:
        driver.execute_script("window.open('"+i+"','_newtab');")
        #לחיצה על כפתור עיצוב אתר
        for handle in driver.window_handles:
            driver.switch_to_window(handle)

        driver.execute_script("Func_MP_ShowPages(this);func_MP_Ifr_Hide();func_MP_LoadPagesList();")
        g = 1
        for i in range(0,len(epi)):
            body = epi[i]
            start = body['start']
            end = body['end']
            prog = body['prog']
            img = body['img']
            for j in range(int(start),int(end)+1):
               post = 86
               print("PAGE",g)
               name = prog + " פרק " + str(j) + " לצפייה ישירה"
               mog = prog + " פרק " + str(j)
               driver.execute_script("SubAddNewPage("+(str(g))+");")
               field = driver.find_element_by_id("PageAddName")
               field.clear()
               field.send_keys(name)
               #האם יתווסף לתפריט עליון
               driver.execute_script('document.getElementById("chkAddPageIsVisibleH").click()')
               driver.execute_script("SubPageAdd();")
               driver.implicitly_wait(10)
               driver.maximize_window()
               #driver.execute_script("window.scrollTo(0, document.body.scrollHeight);")
               print("Content Writing...",end="")
               driver.execute_script("document.elementFromPoint(300, 200).click();")
               driver.switch_to_frame(driver.find_element_by_id("EditorIFrmae"))
               wait(50)
               driver.find_element_by_id("a_SB_Content_Page").click()
               driver.implicitly_wait(8)
               con = getCon(name,mog,img)
               wait(150)
               try:
                   driver.switch_to_frame(driver.find_element_by_xpath('//*[@id="TxtSB_Content_Page_ifr"]'))
                   gondi = "document.body.innerHTML = '" + con + "'";
                   print(gondi)
                   driver.execute_script("document.body.innerHTML = '" + con + "'")
                   print("DONE!")
                   driver.switch_to_default_content()
                   driver.switch_to_frame("EditorIFrmae")
                   driver.execute_script("CK_SaveData();")

               except (NoSuchElementException,StaleElementReferenceException):
                   print('NEXT')
                   penelope = False
               driver.switch_to_default_content()
               g = g + 1
        driver.execute_script("window.close();")
        for handle in driver.window_handles:
            driver.switch_to_window(handle)
#פתיחת משתמשים
if(openUsers == "1"):
    logs = open("logs.txt", "w")
    driver.get("http://google.com")
    for i in range(int(usersNum)):
        driver.execute_script("window.open('http://web.2all.co.il/WP_register.aspx');")
        for handle in driver.window_handles:
            driver.switch_to_window(handle)
        driver.maximize_window()
        username = rand_login()
        password = "123123"
        mail = username+"@gmail.com"
        driver.find_element_by_id("username2").send_keys(username)
        driver.find_element_by_id("password").send_keys(password)
        driver.find_element_by_id("password2").send_keys(password)
        driver.find_element_by_id("email").send_keys(mail)
        driver.execute_script('document.getElementById("chkterms").checked = true')
        driver.execute_script("document.getElementById('btnSubmit').scrollIntoView();")
        driver.implicitly_wait(10)
        driver.find_element_by_id("Form1").submit()
        #driver.find_element_by_xpath("//input[@value='לחץ לסיום ההרשמה לאתר']").click()
        logs.write(username+"|"+password+"|"+mail+"\n")
        driver.delete_all_cookies()
        driver.execute_script("window.close()")

        for handle in driver.window_handles:
            driver.switch_to_window(handle)

    logs.close()