2017-10-06 53 views
0

我是新来的蟒蛇试图使一个代码,这将使我infromation在C盘的可用空间和软件路径的所有脑干的CSV文件CSV插入。获取数据,并使用python

下面

是我想要的代码..

import time 
import csv 
import os 
import socket 
import ctypes 
import platform 
import sys 

timestr = time.strftime("%Y.%m.%d") 

filepath = r"D:\_library\_abcd\test" 
filename = timestr + ".csv" 

machinename = socket.gethostname() 
print machinename 

def Get_free_space_mb(direname): 
    if platform.system() == 'Windows': 
     free_bytes = ctypes.c_ulonglong(0) 
     ctypes.windll.kernel32.GetDiskFreeSpaceExW(ctypes.c_wchar_p(direname),None, None, ctypes.pointer(free_bytes)) 
     print(free_bytes.value/1024/1024) 
     return free_bytes.value/1024/1024 
    else: 
     st = os.statvfs(dirname) 
     return st.f_bavail * st.f_frsize/1024/1024 

Cdrivespace = Get_free_space_mb("C:\\") 

def software1(): 
    if os.path.isdir("C:\Program Files\software1"): 
     return "software1 Exist" 
    else: 
     return "software1 not Exist" 
software_1 = software1() 

def software2(): 
    if os.path.isdir("C:\Program Files\software2"): 
     return "software2 Exist" 
    else: 
     return "Nuke9 not Exist" 
software_2 = software2() 

def software3(): 
    if os.path.isdir("C:\Program Files\software3"): 
     return "software3 Exist" 
    else: 
     return "software3 not Exist" 
software_3 = software3() 

with open(os.path.join(filepath,filename), 'ab+') as file_: 
writer = csv.writer(file_,delimiter=",") 
data = [machinename,Cdrivespace,software1,software2,software3] 
writer.writerows([data]) 

我以CSV得到的输出是: (SUPT5)(10150)(在0x02AF47B0功能软件1)(在0x02AF4A70功能软件2)(函数software3在0x02AF4AB0)

什么期望输出为: (SUPT5)(10150 MB)(软件1存在)(软件2存在)(software3不存在)

我也想CSV文件写入Cdrivespace的价值后写MB。 有什么变化,我可以对代码更容易吗?

+0

建议 - 而不是写3个独立的功能SW1,SW2,SW3,合并成一个和传递参数 –

回答

1

data线试试这个:

data = [machinename, str(Cdrivespace) + 'MB', software_1, software_2, software_3] 

你被引用software1等功能,而不是返回的值(software_1等)。