#!/usr/bin/env python
# -*- encoding: utf-8 -*-
# @Author : bajins https://www.bajins.com
# @File : system_util.py
# @Version: 1.0.0
# @Time : 2019/8/22 9:18
# @Project: windows-wallpaper-python
# @Package: 
# @Software: PyCharm
import ctypes
import gc
import os
import platform
import sys
from subprocess import call

import psutil


def check_version():
    """
    判断python版本
    :return:
    """
    v = sys.version_info
    # v.major 大版本号
    # v.minor 小版本号
    if v.major == 3 and v.minor >= 5:
        return
    print('你当前安装的Python是%d.%d.%d，请使用Python3.6及以上版本' % (v.major, v.minor, v.micro))
    exit(1)


def update_lib():
    """
    更新依赖方式一
    :return:
    """
    from pip._internal.utils.misc import get_installed_distributions

    packages = [dist.project_name for dist in get_installed_distributions()]
    call("pip install --upgrade" + ' '.join(packages), shell=True)


def update_lib_two():
    """
    更新依赖方式二
    :return:
    """
    import pkg_resources

    packages = [dist.project_name for dist in pkg_resources.working_set]
    call("pip install --upgrade" + ' '.join(packages), shell=True)


def get_windows_software():
    """
    从注册表获取Windows系统安装的软件列表
    :return:
    """
    import winreg

    # 需要遍历的两个注册表
    sub_keys = [
        r'SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall',
        r'SOFTWARE\Wow6432Node\Microsoft\Windows\CurrentVersion\Uninstall',
    ]
    software_list = {}

    # 连接注册表根键
    regRoot = winreg.ConnectRegistry(None, winreg.HKEY_LOCAL_MACHINE)
    for sub_key in sub_keys:
        # winreg.KEY_ALL_ACCESS,winreg.KEY_READ,winreg.KEY_WRITE
        keyHandle = winreg.OpenKey(regRoot, sub_key, 0, winreg.KEY_READ)
        # 获取该目录下所有键的个数(0-下属键个数;1-当前键值个数)
        for i in range(winreg.QueryInfoKey(keyHandle)[0]):
            try:
                # 穷举每个键，获取键名
                key_name = winreg.EnumKey(keyHandle, i)
                # 根据获取的键名拼接之前的路径作为参数，获取当前键下所属键的控制
                each_key = winreg.OpenKey(regRoot, f"{sub_key}\\{key_name}", 0, winreg.KEY_READ)
                if winreg.QueryInfoKey(each_key)[1] > 1:
                    # 穷举每个键，获取键名、键值以及数据类型
                    # name, value, type = winreg.EnumValue(each_key, j)
                    # print(name,value,type)
                    DisplayName, REG_SZ = winreg.QueryValueEx(each_key, 'DisplayName')
                    DisplayVersion, REG_SZ = winreg.QueryValueEx(each_key, 'DisplayVersion')
                    software_list[DisplayName] = DisplayVersion
            except WindowsError as e:
                pass
            # finally:
            #     winreg.CloseKey(each_key)
    winreg.CloseKey(keyHandle)
    winreg.CloseKey(regRoot)
    return software_list


def is_admin():
    """
    判断是否为管理员权限
    :return:
    """
    try:
        return ctypes.windll.shell32.IsUserAnAdmin()
    except Exception as e:
        return False


def update_fire_wall(key_name='PublicProfile'):
    """
    修改注册表中防火墙的公网和家用网络的开启和关闭
    :param key_name: PublicProfile StandardProfile
    :return:
    """
    import winreg

    sub_dir = r'SYSTEM\CurrentControlSet\Services\SharedAccess\Parameters\FirewallPolicy'
    # 连接注册表根键
    regRoot = winreg.ConnectRegistry(None, winreg.HKEY_LOCAL_MACHINE)
    # 获取指定目录下键的控制
    keyHandel = winreg.OpenKey(regRoot, f"{sub_dir}\\{key_name}")
    if is_admin():
        # 设置该键的指定键值enableKey为value
        winreg.SetValueEx(keyHandel, 'EnableFirewall', 1, winreg.REG_DWORD, 1)
    else:
        if sys.version_info[0] == 3:
            # 获取管理员权限
            ctypes.windll.shell32.ShellExecuteW(None, "runas", sys.executable, __file__, None, 1)
    # 关闭键的控制
    winreg.CloseKey(keyHandel)
    winreg.CloseKey(regRoot)


def restart_process(path):
    """
    当内存占用达到一定比例进程重启
    :param path: 执行脚本的全路径
    :return:
    """
    if psutil.virtual_memory().percent >= 80:
        print("当前进程id：", os.getpid())
        print("前进程组id：", os.getpgid(), os.getpgrp())
        print("父进程id：", os.getppid())
        print('内存使用：', psutil.Process(os.getpid()).memory_info().rss)
        print("当前内存占用率：", psutil.virtual_memory().percent)
        # if gc.isenabled():
        #     # 释放内存
        #     gc.collect()

        py = "python3" if (os.system("python3 -V") == 0) else "python"
        sysstr = platform.system()
        if sysstr == "Windows":
            os.system(f"taskkill /pid {os.getpid()} /f && {py} {path}")
        elif sysstr == "Linux":
            os.system(f"kill -9 {os.getpid()} && {py} {path}")