import os
import sys
import django

#path=os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
path=os.path.dirname(os.path.dirname(os.path.abspath('.')))

sys.path.insert(0,path)
#os.environ['DJANGO_SETTINGS_MODULE']='ansible_ui.settings'
os.environ['DJANGO_SETTINGS_MODULE']='ansible_ui.settings'
django.setup()

from public.models import *
from django.contrib.auth.models import User
from tools.config import inventory


u,b = User.objects.get_or_create(username='root')
if b:
    u.username = 'root'
    u.is_staff = True
    u.is_superuser = True
    u.set_password('1234567')
    u.save()
    print("\33[34m创建管理员，root: 1234567")
else:
    print('超级管理员已存在')


#pb,b = Functions.objects.get_or_create(playbook='test_debug.yml')
#if b:
#    print('创建测试 playbook')
#    pb.nickName = '测试Debug'
#    pb.playbook = 'test_debug.yml'
#    pb.save()
#else:
#    print('测试 playbook 已存在')
#
#pb,b = Functions.objects.get_or_create(playbook='wordpress.yml')
#if b:
#    print('创建部署wordpress playbook')
#    pb.nickName = '部署wordpress'
#    pb.playbook = 'wordpress.yml'
#    pb.save()
#else:
#    print('测试部署wordpress 已存在')
ps = os.listdir('playbooks/')
for p in ps:
    if os.path.isfile('playbooks/%s' % p):
        with open('playbooks/%s' % p) as f:
            s = f.read()
        sl = s.strip().splitlines()
        if not sl: continue
        t = sl[0]
        if not t.startswith('#') or t[1:].strip().startswith('- hosts'):
            continue
        fs = Functions.objects.get_or_create(playbook=p)
        if fs[1]:
            fs[0].nickName=t[1:].strip()
            fs[0].save()
            print("新PlayBook： %s" % p)
# end insert playbook

h1,b = HostsLists.objects.get_or_create(hostname='测试主机', ip='1.1.1.1')
if b:
    print('添加测试主机')
h2,b = HostsLists.objects.get_or_create(hostname='本机', ip='127.0.0.1')
if b:
    print('添加本机')
g,b = ProjectGroups.objects.get_or_create(groupName='test', nickName='测试')
if b:
    print('添加测试组')
g.hostList.add(h1)
g.hostList.add(h2)
g.save()



data = "# 请勿手动修改该文件\n"
gs = ProjectGroups.objects.all()
for g in gs:
    data += '\n# %s\n[%s]\n' % (g.nickName, g.groupName)
    data += '\n'.join([ i[0] for i in g.hostList.values_list('ip') ])
with open(inventory, 'w') as f:
    f.write(data+ '\n')