import mysql.connector
import requests
import json
import urllib3
from collections import OrderedDict
import sys
import ConfigParser
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
urllib3.disable_warnings()


config = ConfigParser.ConfigParser()
config.read("[Path to your config file]")
testrailUser = config.get("testrail","user")
testrailPassword = config.get("testrail","password")
databaseUser = config.get("mysql","user")
databasePassword = config.get("mysql","password")
testrailUrl = config.get("testrail","url")
connectionMysql = mysql.connector.connect(host="localhost",user=databaseUser,password=databasePassword,database="testrail")
cursorMysql = connectionMysql.cursor()
cursorMysql.execute("SELECT id,plan_id FROM runs WHERE entry_id = '"+sys.argv[1]+"'and plan_id = '"+sys.argv[2]+"'")
result = cursorMysql.fetchall()
for row in result:
    id = row[0]
    plan_id = row[1]
    headers = {"Content-Type": "application/json"}
    r = requests.get(testrailUrl+"get_run/"+str(id),auth=(testrailUser,testrailPassword),headers=headers,verify=False)
    jsonRun = json.loads(r.text)
    name = jsonRun["name"]+"_copied"
    suiteId = jsonRun["suite_id"]
    projectId = jsonRun["project_id"]
    assignedTo = jsonRun["assignedto_id"]
    description = jsonRun["description"]
    cursorMysql.execute("SELECT content_id FROM tests WHERE run_id = "+str(id)+" AND is_selected = 1")
    tests = cursorMysql.fetchall()
    tests = [x[0] for x in tests]
    print tests
    payloadPlan = OrderedDict()
    payloadPlan["suite_id"] = suiteId
    payloadPlan["assignedto_id"] = assignedTo
    payloadPlan["include_all"] = False
    payloadPlan["description"] = description
    payloadPlan["name"] = name
    payloadPlan["case_ids"] = tests
    payload = json.dumps(payloadPlan)
    print payload
    r = requests.post(testrailUrl + "add_plan_entry/"+str(plan_id),data=payload,auth=(testrailUser,testrailPassword),headers=headers,verify=False)
    print r.text
    break