from fabric.api import hosts, run, env, local, cd, get, lcd
from fabric.tasks import execute

env.hosts = ["fab@192.168.1.101:22", "root@192.168.1.101:22"]
env.passwords = {"fab@192.168.1.101:22": "fab", "root@192.168.1.101:22": "tofabor"}


@hosts("ktv@192.168.1.101:22")
def update():
  """更新测试服务器代码"""
  with cd("/opt/project/project"): # 进入测试服务器的项目目录
    run("git pull origin master") # 从git服务器的master分支下拉最新代码
    run("/usr/local/bin/python2.7 /opt/project/project/manage.py makemigrations") # 这是django框架检测数据库变动的命令
    run("/usr/local/bin/python2.7 /opt/project/project/manage.py migrate") # 这是django框架执行数据库变更的命令

@hosts("ktv@192.168.1.101:22")
def restart():
  """重启服务"""
  execute('stop')
  execute('start')


@hosts("root@192.168.1.101:22")
def start():
  """开始服务"""
  with cd("/opt/project/project"):
    run("supervisorctl start dev")

@hosts("ktv@192.168.1.101:22")
def stop():
  """停止服务"""
  pids = run("ps -ef |grep '9001'| awk '{print $2}'")
  pid_list = pids.split('\r\n')
  for i in pid_list[:-2]:
    run('kill -9 %s' % i) # 杀掉运行服务进程