提交代码
This commit is contained in:
parent
b4a512776a
commit
9c24f49d79
|
|
@ -0,0 +1,10 @@
|
|||
changes log
|
||||
======================================
|
||||
|
||||
------------------------------
|
||||
1.0.0 2023-11-20
|
||||
|
||||
1.第一版已经初步完成
|
||||
|
||||
|
||||
[mh]
|
||||
|
|
@ -0,0 +1,5 @@
|
|||
include README.txt
|
||||
include requirements.txt tox.ini
|
||||
recursive-include test *.txt *.py
|
||||
recursive-include *.txt *.py *.tmpl
|
||||
recursive-include bin *.*
|
||||
|
|
@ -0,0 +1,27 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
import os
|
||||
import sys
|
||||
import site
|
||||
|
||||
# set path
|
||||
py_dir = os.path.dirname(os.path.realpath(__file__))
|
||||
|
||||
parent_dir = os.path.dirname(py_dir)
|
||||
|
||||
# print parent_dir
|
||||
# if os.path.isdir(os.path.join(parent_dir, "delegate")):
|
||||
# sys.path.append(parent_dir)
|
||||
|
||||
libdir = os.path.join(parent_dir, "lib")
|
||||
# print libdir
|
||||
if os.path.isdir(libdir):
|
||||
old_len = len(sys.path)
|
||||
new_sys_path = []
|
||||
site.addsitedir(libdir) # @UndefinedVariable
|
||||
for item in sys.path[old_len:]:
|
||||
new_sys_path.append(item)
|
||||
sys.path.remove(item)
|
||||
sys.path[:0] = new_sys_path
|
||||
|
||||
# set cwd
|
||||
os.chdir(parent_dir)
|
||||
|
|
@ -0,0 +1,176 @@
|
|||
#!/usr/bin/python3
|
||||
import psutil
|
||||
import smtplib
|
||||
import os
|
||||
import argparse
|
||||
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
from email.mime.text import MIMEText
|
||||
from email.header import Header
|
||||
from email.mime.multipart import MIMEMultipart
|
||||
|
||||
# 定义公共的部分
|
||||
path = Path(__file__)
|
||||
report_dir = path.parent.parent / "report"
|
||||
# 初始化目录
|
||||
if not report_dir.exists():
|
||||
# shutil.rmtree(data_dir)
|
||||
report_dir.mkdir()
|
||||
#报告文件
|
||||
week=str(datetime.now().isocalendar()[1])
|
||||
file_name = "ComputerInfos-week-" + week + ".txt"
|
||||
file_path = report_dir/file_name
|
||||
|
||||
#发邮件
|
||||
def send_mail():
|
||||
#发件人
|
||||
sender = 'mh@unamail.com'
|
||||
# 接收邮件，可以发给多人
|
||||
receivers = ['mh@unamail.com']
|
||||
# 邮件主体
|
||||
msg = MIMEMultipart()
|
||||
#正文
|
||||
message = "第{0}周，{1}运行信息检测已完成，详情请查看附件！".format(week,get_IP())
|
||||
msg.attach(MIMEText(message, 'plain', _charset="utf-8"))
|
||||
# 发送者
|
||||
msg['From'] = Header(sender, 'utf-8')
|
||||
# 接收者
|
||||
msg['To'] = Header(receivers[0], 'utf-8')
|
||||
# 主题
|
||||
subject = '【长期任务】第{0}周{1}系统运行信息'.format(week,get_IP())
|
||||
msg['Subject'] = Header(subject, 'utf-8')
|
||||
#附件信息
|
||||
att = MIMEText(open(str(file_path), 'rb').read(), 'base64', 'utf-8')
|
||||
att["Content-Type"] = 'application/octet-stream'
|
||||
att["Content-Disposition"] = 'attachment; filename="{}"'.format(file_name)
|
||||
msg.attach(att)
|
||||
|
||||
try:
|
||||
smtpObj = smtplib.SMTP('10.10.110.102')
|
||||
smtpObj.sendmail(sender, receivers, msg.as_string())
|
||||
print ("邮件发送成功")
|
||||
except smtplib.SMTPException:
|
||||
print ("Error: 无法发送邮件")
|
||||
|
||||
|
||||
#写入文件
|
||||
def save_txt(datas):
|
||||
with open(str(file_path),'a+',encoding='utf-8',newline='') as file_txt:
|
||||
[file_txt.write(str(item)+'\n') for item in datas]
|
||||
|
||||
|
||||
# 获取本机磁盘使用率和剩余空间G信息
|
||||
def get_disk_info():
|
||||
# 循环磁盘分区
|
||||
content = []
|
||||
for disk in psutil.disk_partitions():
|
||||
# 读写方式 光盘 or 有效磁盘类型
|
||||
if 'cdrom' in disk.opts or disk.fstype == '':
|
||||
continue
|
||||
disk_name_arr = disk.device.split(':')
|
||||
disk_name = disk_name_arr[0]
|
||||
disk_info = psutil.disk_usage(disk.device)
|
||||
# 磁盘剩余空间，单位G
|
||||
free_disk_size = disk_info.free//1024//1024//1024
|
||||
# 磁盘总空间，单位G
|
||||
total_disk_size = disk_info.total//1024//1024//1024
|
||||
# 当前磁盘使用率、剩余空间G和磁盘总空间信息
|
||||
info = " %s盘使用率：%s%%， 剩余空间：%iG， 总大小：%sG" % (disk_name, str(disk_info.percent),free_disk_size,total_disk_size)
|
||||
# print(info)
|
||||
# 拼接多个磁盘的信息
|
||||
content.append(info)
|
||||
print(content)
|
||||
save_txt(content)
|
||||
|
||||
# 获取某个目录的大小
|
||||
dir_content=[]
|
||||
def get_dir_size(path):
|
||||
list1 = []
|
||||
fileList = os.listdir(path) # 获取path目录下所有文件
|
||||
for filename in fileList:
|
||||
pathTmp = os.path.join(path,filename) # 获取path与filename组合后的路径
|
||||
if os.path.isdir(pathTmp): # 判断是否为目录
|
||||
print("---------------")
|
||||
dir_content.append(" ---------------")
|
||||
get_dir_size(pathTmp) # 是目录就继续递归查找
|
||||
elif os.path.isfile(pathTmp): # 判断是否为文件
|
||||
filesize = os.path.getsize(pathTmp) # 如果是文件，则获取相应文件的大小
|
||||
str_tex = ' %s文件的大小为：%d字节' % (pathTmp,filesize)
|
||||
print(str_tex)
|
||||
dir_content.append(str_tex)
|
||||
list1.append(filesize) # 将文件的大小添加到列表
|
||||
str_dir_tex = ' %s 目录的大小为: %.4f MB' % (path, (sum(list1)/1024/1024))
|
||||
print(str_dir_tex)
|
||||
dir_content.append(str_dir_tex)
|
||||
|
||||
|
||||
|
||||
# cpu信息
|
||||
def get_cpu_info():
|
||||
cpu_percent = psutil.cpu_percent(interval=1)
|
||||
cpu_info = ["","CPU使用率：%i%%" % cpu_percent,""]
|
||||
print(cpu_info)
|
||||
# return cpu_info
|
||||
save_txt(cpu_info)
|
||||
|
||||
# 内存信息
|
||||
def get_memory_info():
|
||||
virtual_memory = psutil.virtual_memory()
|
||||
used_memory = virtual_memory.used/1024/1024/1024
|
||||
free_memory = virtual_memory.free/1024/1024/1024
|
||||
memory_percent = virtual_memory.percent
|
||||
memory_info = ["内存使用：%0.2fG，使用率%0.1f%%，剩余内存：%0.2fG" % (used_memory, memory_percent, free_memory),""]
|
||||
print(memory_info)
|
||||
# return memory_info
|
||||
save_txt(memory_info)
|
||||
|
||||
|
||||
def get_IP():
|
||||
"""获取ipv4地址"""
|
||||
dic = psutil.net_if_addrs()
|
||||
ipv4_list = []
|
||||
for adapter in dic:
|
||||
snicList = dic[adapter]
|
||||
for snic in snicList:
|
||||
# if snic.family.name in {'AF_LINK', 'AF_PACKET'}:
|
||||
# mac = snic.address
|
||||
if snic.family.name == 'AF_INET':
|
||||
ipv4 = snic.address
|
||||
if ipv4 != '127.0.0.1':
|
||||
ipv4_list.append(ipv4)
|
||||
# elif snic.family.name == 'AF_INET6':
|
||||
# ipv6 = snic.address
|
||||
print(ipv4_list)
|
||||
return(ipv4_list[0])
|
||||
|
||||
def main():
|
||||
commoninfo = []
|
||||
if not file_path.exists():
|
||||
t1 = "本周是第{0}周".format(week)
|
||||
commoninfo.append(t1)
|
||||
commoninfo.append("")
|
||||
commoninfo.append("--------------------START--------------------------")
|
||||
commoninfo.append("检测时间： {0}".format(datetime.now()))
|
||||
commoninfo.append("检测IP： {0}".format(get_IP()))
|
||||
save_txt(commoninfo)
|
||||
save_txt(["","磁盘使用情况:",""])
|
||||
get_disk_info()
|
||||
get_cpu_info()
|
||||
get_memory_info()
|
||||
path = "D:\\agent\\ubackup\\uagent\\logs"
|
||||
get_dir_size(path)
|
||||
save_txt(["{0}目录大小:".format(path)])
|
||||
save_txt(dir_content)
|
||||
save_txt(['----------------------END----------------------------'])
|
||||
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
#默认不发邮件，通过参数控制
|
||||
parser = argparse.ArgumentParser(description='send email')
|
||||
parser.add_argument("--send-mail", type=bool, default=False)
|
||||
args = parser.parse_args()
|
||||
if args.send_mail:
|
||||
send_mail()
|
||||
|
|
@ -0,0 +1,7 @@
|
|||
#邮箱收件人,支持多个
|
||||
receivers:
|
||||
- "mh@unamail.com"
|
||||
|
||||
#agent安装路径
|
||||
agent_dir : "D:\\agent\\ubackup\\uagent\\logs"
|
||||
|
||||
|
|
@ -0,0 +1,101 @@
|
|||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
"""
|
||||
create zip 安装包
|
||||
"""
|
||||
|
||||
import logging
|
||||
import os.path
|
||||
import shutil
|
||||
import sys
|
||||
|
||||
|
||||
def _copytree(src, dst, ignore=None):
|
||||
|
||||
names = os.listdir(src)
|
||||
if ignore is not None:
|
||||
ignored_names = ignore(src, names)
|
||||
else:
|
||||
ignored_names = set()
|
||||
try:
|
||||
os.makedirs(dst)
|
||||
except Exception:
|
||||
pass
|
||||
errors = []
|
||||
for name in names:
|
||||
if name in ignored_names:
|
||||
continue
|
||||
srcname = os.path.join(src, name)
|
||||
dstname = os.path.join(dst, name)
|
||||
try:
|
||||
if os.path.isdir(srcname):
|
||||
shutil.copytree(srcname, dstname, ignore=ignore)
|
||||
else:
|
||||
# Will raise a SpecialFileError for unsupported file types
|
||||
shutil.copy2(srcname, dstname)
|
||||
# catch the Error from the recursive copytree so that we can
|
||||
# continue with other files
|
||||
except shutil.Error as err:
|
||||
errors.extend(err.args[0])
|
||||
except EnvironmentError as why:
|
||||
errors.append((srcname, dstname, str(why)))
|
||||
try:
|
||||
shutil.copystat(src, dst)
|
||||
except OSError as why:
|
||||
if WindowsError is not None and isinstance(why, WindowsError):
|
||||
# Copying file access times may fail on Windows
|
||||
pass
|
||||
else:
|
||||
errors.extend((src, dst, str(why)))
|
||||
if errors:
|
||||
raise shutil.Error(errors)
|
||||
|
||||
|
||||
def _zip_file(target_dir):
|
||||
root_dir = os.path.dirname(target_dir)
|
||||
os.chdir(root_dir)
|
||||
shutil.make_archive(os.path.basename(target_dir), format="gztar",
|
||||
base_dir=os.path.basename(target_dir))
|
||||
|
||||
|
||||
def _strip_py(py_dir):
|
||||
for base, dirs, files in os.walk(py_dir):
|
||||
for name in files:
|
||||
if name.endswith('.py'):
|
||||
path = os.path.join(base, name)
|
||||
logging.debug("Deleting %s", path)
|
||||
os.unlink(path)
|
||||
|
||||
|
||||
def main():
|
||||
# src_dir = sys.argv[1]
|
||||
site_pacakge_dir = sys.argv[2]
|
||||
target_dir = sys.argv[3]
|
||||
|
||||
top_dir = sys.argv[4]
|
||||
|
||||
shutil.rmtree(target_dir, ignore_errors=True)
|
||||
os.makedirs(target_dir)
|
||||
|
||||
for dir in ("bin", "etc"):
|
||||
_copytree(os.path.join(top_dir, dir),
|
||||
os.path.join(target_dir, dir))
|
||||
|
||||
shutil.copy2(os.path.join(top_dir, "README.md"),
|
||||
os.path.join(target_dir, "README.md"))
|
||||
|
||||
target_lib_dir = os.path.join(target_dir, "lib")
|
||||
_copytree(site_pacakge_dir, target_lib_dir)
|
||||
|
||||
_zip_file(target_dir)
|
||||
|
||||
print("")
|
||||
print("output dir %s" %(target_dir))
|
||||
|
||||
if __name__ == '__main__':
|
||||
try:
|
||||
main()
|
||||
except Exception:
|
||||
logging.exception("main except")
|
||||
sys.exit(1)
|
||||
|
|
@ -0,0 +1 @@
|
|||
psutil
|
||||
|
|
@ -0,0 +1,32 @@
|
|||
from setuptools import setup, find_packages
|
||||
|
||||
|
||||
install_requires=[]
|
||||
|
||||
for line in open('requirements.txt'):
|
||||
install_requires.append(line.strip())
|
||||
|
||||
setup(name='check_system_info',
|
||||
version='0.0.1',
|
||||
description='',
|
||||
long_description="""\
|
||||
""",
|
||||
# Get more strings from http://www.python.org/pypi?%3Aaction=list_classifiers # nopep8
|
||||
classifiers=[
|
||||
"Programming Language :: Python",
|
||||
],
|
||||
keywords='',
|
||||
author='mh',
|
||||
author_email='menghan@unary.com.cn',
|
||||
url='',
|
||||
license='GPL',
|
||||
packages=find_packages(exclude=["ez_setup","test.*", "test"]),
|
||||
namespace_packages=[],
|
||||
include_package_data=True,
|
||||
test_suite='nose.collector',
|
||||
zip_safe=False,
|
||||
install_requires=install_requires,
|
||||
entry_points="""
|
||||
# -*- Entry points: -*-
|
||||
""",
|
||||
)
|
||||
|
|
@ -0,0 +1,6 @@
|
|||
|
||||
set PROJECT_HOME=%~dp0
|
||||
|
||||
|
||||
rem 确保vscode在path里面
|
||||
start code %PROJECT_HOME%
|
||||
|
|
@ -0,0 +1,46 @@
|
|||
[tox]
|
||||
envlist = devenv
|
||||
minversion = 1.6
|
||||
skipsdist = False
|
||||
|
||||
[testenv]
|
||||
install_command = pip install --force-reinstall -U {opts} {packages}
|
||||
setenv = VIRTUAL_ENV={envdir}
|
||||
NOSE_WITH_COVERAGE=1
|
||||
NOSE_COVER_BRANCHES=1
|
||||
deps =
|
||||
-r{toxinidir}/requirements.txt
|
||||
|
||||
|
||||
[testenv:devenv]
|
||||
envdir = devenv
|
||||
basepython = python3.8
|
||||
usedevelop = True
|
||||
|
||||
|
||||
|
||||
[testenv:devenv2]
|
||||
envdir = devenv2
|
||||
basepython = python2.7
|
||||
usedevelop = True
|
||||
|
||||
|
||||
[testenv:py38-release]
|
||||
basepython = python3.8
|
||||
deps =
|
||||
-r{toxinidir}/requirements.txt
|
||||
|
||||
commands =
|
||||
{envpython} {toxinidir}/release.py {envdir} {envsitepackagesdir} {toxinidir}/build/check_system_info_v1.0.0 {toxinidir}
|
||||
|
||||
|
||||
|
||||
|
||||
[testenv:py27-release]
|
||||
basepython = python2.7
|
||||
deps =
|
||||
-r{toxinidir}/requirements.txt
|
||||
|
||||
commands =
|
||||
{envpython} {toxinidir}/release.py {envdir} {envsitepackagesdir} {toxinidir}/build/check_system_info_v1.0.0 {toxinidir}
|
||||
|
||||
Loading…
Reference in New Issue