#!/usr/bin/env python

# -*- encoding: utf-8 -*-

#

# @Description:

# @PreInstall: Pillow

# @Author : bajins https://www.bajins.com

# @File : file_util.py

# @Version: 1.0.0

# @Time : 2019/8/21 15:32

# @Project: windows-wallpaper-python

# @Package:

# @Software: PyCharm

import configparser

import os

import stat

import time

import zipfile

from shutil import copy

# pip install Pillow

from PIL import Image

from . import string_util

def path_join(*path):

"""

路径拼接

:param path:路径字符串数组

:return:

"""

final_path = ""

for i in range(len(path)):

p = path[i]

if string_util.is_empty(p):

continue

if string_util.check_startswith(p):

p = p[1:]

if string_util.check_endswith(p):

p = p[:-1]

if i == 0:

final_path = p

else:

final_path = os.path.join(final_path, p)

def image_to_bmp(image_path):

"""

转换图片为bmp格式

:param image_path:

:return:

"""

# 分割路径和文件名

filepath, filename = os.path.split(image_path)

# 分割文件的名字和后缀

filename, extension = os.path.splitext(filename)

# 替换文件后缀组成新的路径

new_path = image_path.replace(extension, '.bmp')

# 打开图片

bmp_image = Image.open(image_path)

# 保存为bmp

bmp_image.save(new_path, "BMP")

return new_path

def replace_file_content(file, old_str, new_str):

"""

替换文件中的字符串

:param file:文件名

:param old_str:旧字符串

:param new_str:新字符串

:return:

"""

file_data = ""

with open(file, "r", encoding="utf-8") as f:

for line in f:

if old_str in line:

line = line.replace(old_str, new_str)

file_data += line

with open(file, "w", encoding="utf-8") as f:

f.write(file_data)

def zip_extract(file_path, pwd):

"""

解压zip文件

:param file_path: zip文件路径

:param pwd: 解压目的地目录

:return:

"""

zip_file = zipfile.ZipFile(file_path, "r")

# ZipFile.namelist(): 获取ZIP文档内所有文件的名称列表

for fileM in zip_file.namelist():

zip_file.extract(fileM, pwd)

zip_file.close()

def parent_path(file):

"""

获取文件的父级目录

:param file:文件

:return:

"""

return os.path.dirname(os.path.dirname(file))

def remove_read_only(filename):

"""

清除文件的只读标记

stat.S_IREAD: windows下设为只读

stat.S_IWRITE: windows下取消只读

stat.S_IROTH: 其他用户有读权限

stat.S_IRGRP: 组用户有读权限

stat.S_IRUSR: 拥有者具有读权限

:param filename:

:return:

"""

os.chmod(filename, stat.S_IWRITE)

def read_file(file_path):

"""

读取文件内容

:param file_path: 文件全路径

:return:

"""

# 一次性读入txt文件，并把内容放在变量lines中

with open(file_path) as lines:

# 返回的是一个列表，该列表每一个元素是txt文件的每一行

return lines.readlines()

def read_file_remove_line_feed(file_path):

"""

读取文件内容并删除换行符

:param file_path: 文件全路径

:return:

"""

# 一次性读入txt文件，并把内容放在变量lines中

with open(file_path) as lines:

# 返回的是一个列表，该列表每一个元素是txt文件的每一行

array = lines.readlines()

# 使用一个新的列表来装去除换行符\n后的数据

array2 = []

# 遍历array中的每个元素

for i in array:

# 去掉换行符\n

i = i.strip('\n')

# 把去掉换行符的数据放入array2中

array2.append(i)

return array2

def write_temp(file_path, lines):

"""

创建临时文件 import tempfile

:param file_path:文件全路径

:param lines:内容

:return:

"""

with open(file_path, 'wt') as f:

f.writelines(lines)

return f.name

def write_lines(file_path, lines):

"""

覆盖文件内容，在文件中写入多行

:param file_path: 文件全路径

:param lines: 写入内容数组

:return:

"""

with open(file_path, "w+") as f:

f.writelines(lines)

f.close()

def delete_size(min_size):

"""

删除小于指定值的文件（单位：K）

:param min_size:

:return:

"""

# 列出目录下的文件

files = os.listdir(os.getcwd())

for file in files:

if os.path.getsize(file) < min_size * 1000:

# 删除文件

os.remove(file)

print(file + " deleted")

return

def delete_null_file():

"""

删除所有大小为0的文件

:return:

"""

files = os.listdir(os.getcwd())

for file in files:

# 获取文件大小

if os.path.getsize(file) == 0:

os.remove(file)

print(file + " deleted.")

return

def create_file(suffix):

"""

根据本地时间创建指定后缀的新文件，如果已存在则不创建

:param suffix: 后缀

:return:

"""

# 将指定格式的当前时间以字符串输出

t = time.strftime('%Y-%m-%d', time.localtime())

new_file = t + suffix

if not os.path.exists(new_file):

f = open(new_file, 'w')

print(new_file)

f.close()

print(new_file + " created.")

else:

print(new_file + " already existed.")

class Config:

def __init__(self, filename):

"""

配置初始化

:param filename:配置文件全路径

"""

self.filename = filename

def read(self):

"""

获取配置文件

:return:

"""

if self.filename == "" or self.filename is None:

raise ValueError("请输入正确的配置文件名！")

if not os.path.exists(self.filename):

raise ValueError("配置文件不存在！")

config = configparser.ConfigParser()

config.read(self.filename)

return config

def sections(self):

"""

获取配置组名

:return:

"""

return self.read().sections()

def get(self, section, key=None):

"""

获取配置值

:param section: 配置组名称

:param key: 配置组中的配置名

:return:

"""

if section == "" or section is None:

raise ValueError("配置组名不能为空！")

if key != "" and key is not None:

return self.read()[section][key]

return self.read()[section]

def count_dir_size(dir_path):

"""

获取目录大小

:param dir_path: 目录

:return:

"""

size = 0

for root, dirs, files in os.walk(dir_path):

size += sum([os.path.getsize(os.path.join(root, name)) for name in files])

return size

def size_unit_format(size, is_speed=False, precision=2):

"""

文件大小自动转换

byte ---- (B)

kilobyte ---- (KB)

megabyte ---- (MB)

gigabyte ---- (GB)

terabyte ---- (TB)

petabyte ---- (PB)

exabyte ---- (EB)

zettabyte ---- (ZB)

yottabyte ---- (YB)

:param size: 大小

:param is_speed: 是否为传输速率计算(bps/bit)

:param precision: 精确到小数点位数

:return:

"""

if not (isinstance(size, float) or isinstance(size, int)):

raise TypeError('需要浮点数或整数！')

if size <= 0:

raise ValueError('数字必须大于零')

formats = ['KB', 'MB', 'GB', 'TB', 'PB', 'EB', 'ZB', 'YB']

unit = 1000.0 if is_speed else 1024.0

for i in formats:

size /= unit

if size < unit:

return f'{round(size, precision)}{i}'

return f'{round(size, precision)}{i}'

def copy_dir(dir, newdir):

"""

复制目录到指定位置

import shutil

shutil.copytree(user_data, mkdtemp, True)

import distutils.dir_util

distutils.dir_util.copy_tree(user_data, mkdtemp)

:param dir: 需拷贝的文件夹

:param newdir: 是拷贝的地方

:return:

"""

for p in os.listdir(dir):

filepath = os.path.join(newdir, p)

old_path = os.path.join(dir, p)

if os.path.isdir(old_path):

os.mkdir(filepath)

copy_dir(old_path, filepath)

if os.path.isfile(old_path):

copy(old_path, filepath)