from selenium.webdriver.common.keys import Keys
import webbrowser
import re
import os
import json
import requests
import random
import selenium
from robobrowser import RoboBrowser
import time
 
class vk_captcha:
    """Класс для работы с каптчей. Вырезает img со страницы, отправляет
    в antigate, получает результат, возвращает введенную каптчу в виде строки"""
 
    def decode(page, root_path):
        """Объединение функций _send_captcha и _check_captcha"""
        print("Каптча...")
 
        key = "4517fb23f100hvyg87gy87gvhj"
        a_captchaID = vk_captcha._send_captcha(page, root_path, key) #antigate captcha ID [OK, 406704123]
        if a_captchaID[0] != "OK":
            print("Неудалось отправить запрос на antigate\n%s" % a_captchaID)
            return 0
        while True:
            time.sleep(7)
            a_captchaRESULT = vk_captcha._check_captcha(a_captchaID[1], key)  #antigate captcah RESULT [OK, Fgv4Kl]
            if a_captchaRESULT[0] != "OK":
                continue
            print("Готово. %s" % a_captchaRESULT[1])
            return a_captchaRESULT[1]
       
 
    def _send_captcha(page, root_path, key):
        """Получает страницу, возвращает массив формата [OK, 406704123]"""
        img_tag = page.find(id="captcha")
        if not img_tag:
            img_tag = page.find("img", {"class":"captcha_img"})
        if not img_tag:
            print("Тега с id=captcha не найдено")
            return 0
        img = vk_captcha.request_with_retry(root_path + img_tag["src"]).content
        data = {
            "key": key,
            "method": "post",
        }
        response = vk_captcha.request_with_retry('http://antigate.com/in.php', data=data, files={"file": img})
        return (response.text.split("|"))
 
    def _check_captcha(captcha_id, key):
        response = vk_captcha.request_with_retry("http://antigate.com/res.php?key=" + key + "&action=get&id=" + captcha_id)
        return (response.text.split('|'))
 
    def request_with_retry(url, data="", files=""):
        TIMEOUT = 0.5
        while True:
            try:
                if data or files:
                    response = requests.post(url, data=data, files=files, timeout=TIMEOUT)
                else:
                    response = requests.get(url, timeout=TIMEOUT)
            except requests.exceptions.ReadTimeout:
                continue
            break
        return response

def create_group(wd):
        with open("file.txt", "r") as f:
            line = f.readline()
            for i, line in enumerate(f):
                if i == 10:
                    break
                else:
                    try:
                        wd.get('http://vk.com/' + line)
                        wd.find_element_by_xpath("//a[@id='profile_photo_link']/img").click()
                        # time.sleep(4)
                        # wd.find_element_by_id("pv_like_count").click()
                    except:
                       
            f.close()