#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
 @FileName:    diskinfo.py
 @Function:    使用subprocess执行shell命令获取阿里云、腾讯云机器磁盘信息
 @Author:      Zhihe An
 @Site:        https://chegva.com
 @Time:        2021/8/29
"""
 
from threading import Thread
import subprocess
import queue
import time
 
num_threads=3
q = queue.Queue()
host_list = []
with open('tmp', 'r') as f:
    lines = f.readlines()
    for l in lines:
        l = l.strip('\n').strip()
        host_list.append(l)

def diskinfo(host, queue):
   while True:
      host = queue.get()
      # ret=subprocess.call('tccli cvm DescribeInstances  --Filters \'[{"Name":"instance-name","Values":["%s"]}]\' --filter "InstanceSet[*]"|jq -r \'.[].DataDisks[0]["DiskSize"],.[].DataDisks[0]["DiskType"]\' | xargs;' % ip,shell=True,stdout=open('result','a'),stderr=subprocess.STDOUT)
      file = open('disk.txt', "a")
      if str(host).startswith('ali'):
         ins_name = subprocess.run('aliyun ecs DescribeInstances --RegionId cn-beijing --InstanceName "%s" | jq -r \'.[].Instance[0]["InstanceId"]\'' % host, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=False)
         # bytes转成字符串
         ins_id = (ins_name.stdout.decode('utf-8').strip('\n').strip())
         # print(ins_id)
         time.sleep(0.5)
         if ins_id != 'null':
            disk_res = subprocess.run('aliyun ecs DescribeDisks --RegionId cn-beijing --InstanceId "%s" --DiskType data |jq -r \'.[]|.["Disk"][0]["Size"],.[][0]["Category"]\' | xargs' % ins_id, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=False)
            disk_size_category = disk_res.stdout.decode('utf-8').strip('\n').strip()
         else:
            disk_size_category = ""
         print(host, disk_size_category)
         print(host, disk_size_category, file=file)
         # file.write(host + "\t" + disk_size_category + "\n")
         time.sleep(0.5)
      elif str(host).startswith('qc'):
         qc_res = subprocess.run('tccli cvm DescribeInstances  --Filters \'[{"Name":"instance-name","Values":["%s"]}]\' --filter "InstanceSet[*]"|jq -r \'.[].DataDisks[0]["DiskSize"],.[].DataDisks[0]["DiskType"]\' | xargs;' % host, shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE, check=False)
         qc_disk = qc_res.stdout.decode('utf-8').strip()
         if qc_disk:
            print(host, qc_disk)
            print(host, qc_disk, file=file)
         else:
            print(host, qc_disk)
            # print(host, qc_disk, file=file)
            file.write("\n")
         # file.write(host + "\t" + qc_disk + "\n")
         time.sleep(0.5)
      else:
         print(host)
         # print(host, file=file)
         file.write("\n")
      file.close()
      queue.task_done()
 
#start num_threads threads
for i in range(num_threads):
   t=Thread(target=diskinfo,args=(i,q))
   t.setDaemon(True)
   t.start()

print('main thread waiting...')
for host in host_list:
   q.put(host)
   q.join()
print('Done')