103 lines
3.1 KiB
Python
103 lines
3.1 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
import os
|
||
|
|
import socket
|
||
|
|
import urllib.request
|
||
|
|
from urllib.error import URLError
|
||
|
|
from multiprocessing import Process
|
||
|
|
import time
|
||
|
|
import platform
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
PATH = lambda p: os.path.abspath(
|
||
|
|
os.path.join(os.path.dirname(__file__), p)
|
||
|
|
)
|
||
|
|
import threading
|
||
|
|
|
||
|
|
|
||
|
|
class AppiumServer:
|
||
|
|
def __init__(self, kwargs=None):
|
||
|
|
self.kwargs = kwargs
|
||
|
|
|
||
|
|
def start_server(self):
|
||
|
|
"""start the appium server
|
||
|
|
"""
|
||
|
|
for i in range(0, len(self.kwargs)):
|
||
|
|
cmd = "appium --session-override -p %s -bp %s -U %s" % (
|
||
|
|
self.kwargs[i]["port"], self.kwargs[i]["bport"], self.kwargs[i]["devices"])
|
||
|
|
print(cmd)
|
||
|
|
if platform.system() == "Windows": # windows下启动server
|
||
|
|
t1 = RunServer(cmd)
|
||
|
|
p = Process(target=t1.start())
|
||
|
|
p.start()
|
||
|
|
while True:
|
||
|
|
print("--------start_win_server-------------")
|
||
|
|
if self.win_is_runnnig("http://127.0.0.1:" + self.kwargs[i]["port"] + "/wd/hub" + "/status"):
|
||
|
|
print("-------win_server_ 成功--------------")
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
appium = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, bufsize=1,
|
||
|
|
close_fds=True)
|
||
|
|
while True:
|
||
|
|
appium_line = appium.stdout.readline().strip().decode()
|
||
|
|
time.sleep(2)
|
||
|
|
print("---------start_server----------")
|
||
|
|
if 'listener started' in appium_line or 'Error: listen' in appium_line:
|
||
|
|
print("----server启动成功---")
|
||
|
|
break
|
||
|
|
|
||
|
|
def win_is_runnnig(self, url):
|
||
|
|
"""Determine whether server is running
|
||
|
|
:return:True or False
|
||
|
|
"""
|
||
|
|
response = None
|
||
|
|
time.sleep(1)
|
||
|
|
try:
|
||
|
|
response = urllib.request.urlopen(url, timeout=5)
|
||
|
|
|
||
|
|
if str(response.getcode()).startswith("2"):
|
||
|
|
return True
|
||
|
|
else:
|
||
|
|
return False
|
||
|
|
except URLError:
|
||
|
|
return False
|
||
|
|
except socket.timeout:
|
||
|
|
return False
|
||
|
|
finally:
|
||
|
|
if response:
|
||
|
|
response.close()
|
||
|
|
|
||
|
|
def stop_server(self, devices):
|
||
|
|
sysstr = platform.system()
|
||
|
|
|
||
|
|
if sysstr == 'Windows':
|
||
|
|
os.popen("taskkill /f /im node.exe")
|
||
|
|
else:
|
||
|
|
for device in devices:
|
||
|
|
# mac
|
||
|
|
cmd = "lsof -i :{0}".format(device["port"])
|
||
|
|
plist = os.popen(cmd).readlines()
|
||
|
|
plisttmp = plist[1].split(" ")
|
||
|
|
plists = plisttmp[1].split(" ")
|
||
|
|
# print plists[0]
|
||
|
|
os.popen("kill -9 {0}".format(plists[0]))
|
||
|
|
|
||
|
|
def re_start_server(self):
|
||
|
|
"""reStart the appium server
|
||
|
|
"""
|
||
|
|
# self.stop_server()
|
||
|
|
# self.start_server()
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class RunServer(threading.Thread):
|
||
|
|
def __init__(self, cmd):
|
||
|
|
threading.Thread.__init__(self)
|
||
|
|
self.cmd = cmd
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
os.system(self.cmd)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
|
||
|
|
pass
|