파이썬으로 서버 hdd 상태 확인해서 mysql 올리기

OS : ubuntu 14.04 LTS

#!/usr/bin/env python
# -*- coding: utf-8 -*-
 
import sys
import subprocess
import commands
import time
import mysql.connector # apt-get install python-mysql.connector
import socket
 
'''
[ smartctl 설치 ]
apt-get install smartmontools 

[ mysql 테이블 ]
CREATE TABLE `hdd`
(
    `idx` BIGINT(20) UNSIGNED NOT NULL AUTO_INCREMENT,
    `host` VARCHAR(64) NOT NULL DEFAULT '',
    `device` VARCHAR(64) NOT NULL DEFAULT '',
    `stat` VARCHAR(64) NOT NULL DEFAULT '',
    `insertDateTime` DATETIME NOT NULL DEFAULT '0000-00-00 00:00:00',
    PRIMARY KEY (`idx`),
    UNIQUE INDEX `idx_UNIQUE` (`idx` ASC)
) ENGINE = InnoDB;

CREATE TABLE `hddMin`
(
    `idx` BIGINT(20) UNSIGNED NOT NULL AUTO_INCREMENT,
    `host` VARCHAR(64) NOT NULL DEFAULT '',
    `device` VARCHAR(64) NOT NULL DEFAULT '',
    `stat` VARCHAR(64) NOT NULL DEFAULT '',
    `insertDateTime` DATETIME NOT NULL DEFAULT '0000-00-00 00:00:00',
    PRIMARY KEY (`idx`),
    UNIQUE INDEX `idx_UNIQUE` (`idx` ASC)
) ENGINE = InnoDB;

'''
 
'''
중복실행 방지코드
'''
isRun = commands.getstatusoutput('ps -aux | grep "' + __file__ + '" | grep -v "grep" | awk "{print \$2}" | wc -l')
 
if int(isRun[1]) > 1:
    print "already run"
    sys.exit()
 
'''
라이브러리
'''
 
def mysqlConnect():
    if hasattr(mysqlConnect, 'conn') == False or mysqlConnect.conn.is_connected() == False:
        mysqlConnect.conn = mysql.connector.connect(user='id', password='pw', host='host', database='server')
        # mysqlConnect.conn = mysql.connector.connect(user='id', password='pw', host='192.168.0.22', database='server')
        mysqlConnect.cursor = mysqlConnect.conn.cursor(buffered=True)
 
    return mysqlConnect
 
def mysqlQuery(queryStr):
    print queryStr
 
    mysqlInit = mysqlConnect();
    mysqlInit.cursor.execute(queryStr)
    mysqlInit.conn.commit()
 
    return mysqlInit.cursor
 
 
def mysqlQueryFetchOne(queryStr):
 
    cursor = mysqlQuery(queryStr)
 
    fetch = cursor.fetchone()
 
    if fetch == None:
        return None
 
    desc = cursor.description
 
    returnArr = {}
 
    for (name, value) in zip(desc, fetch):
        returnArr[name[0]] = value
 
    return returnArr
 
 
def strArrConcat(strArr):
    from cStringIO import StringIO
 
    strIo = StringIO()
 
    for string in strArr:
        strIo.write(str(string))
 
    return strIo.getvalue()
 
 
 
'''
환경변수
'''
envHostName = socket.gethostname()
envNow = time.localtime()
envDateTime = strArrConcat([
    envNow[0]
    ,'-'
    ,envNow[1]
    ,'-'
    ,envNow[2]
    ,' '
    ,envNow[3]
    ,':'
    ,envNow[4]
    ,':00'
    #,envNow[5]
])
 
 
'''
smartctl을 이용해 하드디스크 검사함
'''
smartctlScan = commands.getstatusoutput('smartctl --scan-open')
smartctlScanResult = smartctlScan[1];
smartctlScanArr = smartctlScanResult.split('\n')
 
smartctlCheckDic = {}
smartctlCheckDic.clear()
 
for targetStr in smartctlScanArr:
    if targetStr[:1] != '#':
        target = targetStr[:targetStr.find('#')].strip()
 
        smartctlCheck = commands.getstatusoutput('smartctl -H ' + target)
 
        print smartctlCheck[1]
 
        if smartctlCheck[1].find('failed') > 0:
            smartctlCheckDic[target] = "failed"
 
        elif smartctlCheck[1].find('OK') > 0:
            smartctlCheckDic[target] = "OK"
 
        elif smartctlCheck[1].find('PASSED') > 0:
            smartctlCheckDic[target] = "PASSED"
 
        elif smartctlCheck[1].find('FAILED') > 0:
            smartctlCheckDic[target] = "FAILED"
 
        else:
            smartctlCheckDic[target] = "unknown"
 
print smartctlCheckDic
 
for smartctlCheckDevice in smartctlCheckDic:
    smartctlCheckResult = smartctlCheckDic[smartctlCheckDevice]
 
    mysqlQuery(strArrConcat([
        "DELETE FROM"
        ," `hdd` "
 
        ," WHERE "
        ,"`host` = '"
        ,envHostName
        ,"'"
        ," AND `device` = '"
        ,smartctlCheckDevice
        ,"'"
    ]))
 
    mysqlQuery(strArrConcat([
        "INSERT INTO"
        ," `hdd` "
 
        ," SET "
        ,"`host` = '"
        ,envHostName
        ,"'"
        ,", `device` = '"
        ,smartctlCheckDevice
        ,"'"
        ,",`stat` = '"
        ,smartctlCheckResult
        ,"'"
        ,",`insertDateTime` = '"
        ,envDateTime
        ,"'"
    ]))
 
    mysqlQuery(strArrConcat([
        "INSERT INTO"
        ," `hddMin` "
 
        ," SET "
        ,"`host` = '"
        ,envHostName
        ,"'"
        ,", `device` = '"
        ,smartctlCheckDevice
        ,"'"
        ,",`stat` = '"
        ,smartctlCheckResult
        ,"'"
        ,",`insertDateTime` = '"
        ,envDateTime
        ,"'"
    ]))

cron 에 1분마다 동작하게 등록하면

요래됨 ㅎㅎ