import sys
import codecs
import traceback
import requests
import re
import pandas as pd
import random
import urllib
import json
import gspread
import datetime
from gspread_pandas import Spread, Client
from oauth2client.service_account import ServiceAccountCredentials
import os
import redis
import time
import fire
import dataset


db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/seo?charset=utf8mb4')
lst=[]
cursor = db.query('SELECT query FROM seo.hhh_gsc_imp where position >=2.5 and position <=8.5 order by rand();')

for c in cursor:
    lst.append([c['query']])


def run_once(pport, dockername):
    global lst
#    seeds=['室內 設計','室內設計','客廳 裝潢','裝潢','房間 設計','台北','台北 設計','台北 室內','台北  室內 設計','電視 牆','電視','裝潢 設計','裝潢 室內','室內 設計 費用','設計 費用','居家 空間 設計','居家 設計','居家 空間','空間 設計','衣魚','玄關設計','房間佈置','量子電視','小坪數裝潢','踢腳板','清潔劑','班傑明油漆','冰箱除臭']
#    lst=[['衣櫃設計'],['磁磚尺寸'],['幸福空間設計'],['養魚風水'],['房間風水'],['衣魚'],['冰箱除臭'],['大腸鏡檢查費用'],['房間佈置'],['潮濕蟲漂白水'],['小坪數裝潢'],['幸福空間'],['衣櫃設計'],['門對門'],['玄關設計'],['幸福空間'],['幸福空間'],['衣魚有害嗎'],['幸福空間'],['幸福空間 稅']]
#    lst=[['衣櫃設計'],['磁磚尺寸'],['幸福空間設計'],['養魚風水'],['房間風水'],['衣魚'],['冰箱除臭'],['大腸鏡檢查費用'],['房間佈置'],['潮濕蟲漂白水'],['小坪數裝潢'],['幸福空間'],['衣櫃設計'],['門對門'],['玄關設計'],['幸福空間'],['幸福空間'],['衣魚有害嗎'],['幸福空間'],['幸福空間 稅']]

#    for s in seeds:
#        lst.append([s+' 幸福'])
#        lst.append([s+' 幸福空間'])
#       lst.append([s+' 幸福 空間'])

#    lst = [['幸福空間'],['開運植物  幸福空間'],['幸福 文化美藏'],['三井牙醫  幸福空間'],['現代混搭 幸福空間'],['簡單的事 幸福空間'],['古典風雅 幸福空間'],['木藝 幸福空間']]
#    lst = [['幸福空間'],['開運植物  幸福空間'],['幸福 文化美藏'],['三井牙醫  幸福空間'],['現代混搭 幸福空間'],['簡單的事 幸福空間'],['古典風雅 幸福空間'],['木藝 幸福空間']]


#    cursor = db.query('SELECT * FROM hhh.seo_designer order by rand() limit 40')

    obj = random.choice(lst)
    print(obj)
    kw = obj[0]
    domain = 'hhh.com.tw'
    print(kw,domain)

    s = f'python3 general_clickbot_hhh.py get --kw="{kw}" --domain="{domain}" --port={str(pport)}'

    intval = os.system(f'python3 general_clickbot_hhh.py get --kw="{kw}" --domain="{domain}" --port="{str(pport)}"')
    
    print('執行genetal_clickbot')

    if intval == -1:
        print('-1')
        sys.exit()


class JParams(object):
    
    def get(self, port=9222):
        while True:
            try:
                os.system('docker container restart tiny4')
                os.system('docker container restart tiny5')
                os.system('docker container restart tiny6')
                time.sleep(1)
                run_once(9934, 'tiny4')
                run_once(9935, 'tiny5')
                run_once(9936, 'tiny6')
                time.sleep(20)

            except:
                os.system('docker container restart tiny4')
                time.sleep(15)


if __name__ == '__main__':
    fire.Fire(JParams)