如何从python中保存mongodb中的数据

时间:2013-07-08 11:01:54

标签: django mongodb mongoengine

我的任务是制作像服务器密度这样的服务器监控工具。我有服务器列表,我想要每15秒钟每个服务器的CPU信息,RAM信息,磁盘使用等,并打印其图。 我在django中设置项目,mongo作为后端DB。 我写了python脚本,可以在本地系统上给我这些参数。 现在我的任务是将此信息存储在Mongo-DB中。我已经为每种参数编写了相应的模型。 这是我创建的模型。

class CpuUsage(Document):

    server = ReferenceField(ServerInfo, db_field='se', required=False)
    sys_time = DateTimeField(db_field='st', required=False)
    #CPU=
    usr = StringField(db_field='u', required=False)
    nice = StringField(db_field='n', required=False)
    sys = StringField(db_field='sy', required=False)
    iowait = StringField(db_field='io', required=False)
    irq = StringField(db_field='ir', required=False)
    soft = StringField(db_field='so', required=False)
    steal = StringField(db_field='st', required=False)
    guest = StringField(db_field='g', required=False)
    idle = StringField(db_field='id', required=False)
    meta = {
        'indexes': ['server']
    }

我的脚本有以下两个功能 1)cpu_info将cpu参数提取到字典中。

def cpu_info():
    cpu_parameters = {}
    ram_parmeters = {}
    p_info = '/home/bhavuk/Desktop/p_info'
    #p_file = open(p_info, 'w')
    cmd1 = 'mpstat -P ALL >'+ p_info
    p = subprocess.Popen(cmd1,
                         shell=True,
                         stdout=subprocess.PIPE,
                         stderr=subprocess.STDOUT)
    cpu_cores=open(p_info, 'r').read().split('\n')
    #for i in range(0, len(cpu_cores)):
    #    print i, cpu_cores[i]
    print "current cpu state:\n"
    for i in range(2,len(cpu_cores)-1):
        cpu_parameters[i-2] = cpu_cores[i].split('\t')
        print cpu_parameters[i-2]
    save_cpu_info(cpu_parameters)

2)这将字典中包含的信息保存到mongodb。

def save_cpu_info(parameters):
    s = ServerInfo() #need to correct
    s.save()

    for i in xrange(0, len(parameters)):
        c = CpuUsage(server=s)
        x = parameters[i]

    c.sys_time = x[0]
    c.CPU_core = x[2]
    c.usr = x[3]
    c.nice = x[4]
    c.sys = x[5]
    c.iowait = x[6]
    c.irq = x[7]
    c.soft = x[8]
    c.steal = x[9]
    c.guest = x[10]
    c.idle = x[11]
    c.save()

我对serverfield感到困惑,我应该如何使用它,我的目的是在每个searver上运行这个脚本,并设置一个UI,它可以显示有关每个服务器的信息。

我想知道我正在朝着正确的方向前进,以及如何在每个服务器上以15秒的间隔运行此脚本并绘制图表。

进一步改善将受到高度赞赏。 我也愿意

1 个答案:

答案 0 :(得分:0)

1.要按计划的脚本运行,您可以使用crontab,Celery或     自己的守护进程。看来crontab会更合适。你必须这样做     记得使用crontab选择你应该制作控制台     srcript as:

#!/usr/bin/env python2.7
   # import required packages
   # setup paths to django project
   os.environ.setdefault("DJANGO_SETTINGS_MODULE", "your_project.settings")
   sys.path.append(('/path/to/django/project'))
   ... # your code


if __name__ == "__main__":

   ... # your code

2.如果您想将数据保存在一个地方,您应该只使用一个共同的主机     MongoDB。这意味着您应该使用与您的服务器相同的ip / domain     在settings.py中将MongoDB改为localhost到每个Django中     项目