0%

Python小技巧

语法检查

pylint 是一个能够检查Python编码质量、编码规范的工具。它分析 Python 代码中的错误,查找不符合代码风格标准(Pylint 默认使用的代码风格是 PEP 8)和有潜在问题的代码。

个人认为正确性比风格更加重要,不妨大材小用,执行脚本之前,都使用pylint进行语法检查一下。

1
2
pip install pylint
pylint test.py

导入自定义模块

全局导入

自己新建了一个模块,怎样让它可以被全局引用?答:导入自定义模块。
具体操作方法:
1、进入 xxx/python3/lib/python3.6/site-packages 目录
2、新建 yyy.pth 文件,写入自定义模块的路径

1
/home/voidking/scripts/vktools/

详情参考python之使用.pth文件导入自定义模块

指定文件导入

自己新建了一个模块a,路径不同的情况下,怎样让它被模块b引用?答:使用 sys.path.append 函数。

1
2
# sys.path.remove('/path/to/a')
sys.path.append('/path/to/a')

获取脚本路径

已知路径:

1
2
3
4
5
path
└── module_a
├── getpath.py
└── module_b
└── getpath.py

getpath.py脚本内容为:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import os
import sys
from module_b import getpath

def print_path():
print(f'os.getcwd() is {os.getcwd()}')
print(f'sys.path[0] is {sys.path[0]}')
print(f'sys.argv[0] is {sys.argv[0]}')
print(f'__file__ is {__file__}')
print(f'os.path.dirname(__file__) is {os.path.dirname(__file__)}')
print(f'os.path.abspath(__file__) is {os.path.abspath(__file__)}')
print(f'os.path.realpath(__file__) is {os.path.realpath(__file__)}')
print(f'os.path.split(os.path.realpath(__file__))[0] is {os.path.split(os.path.realpath(__file__))[0]}')

print('------ module_a/getpath.py ------')
print_path()
print('---------------------------------')
print('------ module_a/module_b/getpath.py ------')
getpath.print_path()
print('------------------------------------------')

在path目录中执行脚本python module_a/getpath.py ,得到结果为:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
------ module_a/getpath.py ------
os.getcwd() is /Users/vk/tmp/path
sys.path[0] is /Users/vk/tmp/path/module_a
sys.argv[0] is module_a/getpath.py
__file__ is module_a/getpath.py
os.path.dirname(__file__) is module_a
os.path.abspath(__file__) is /Users/vk/tmp/path/module_a/getpath.py
os.path.realpath(__file__) is /Users/vk/tmp/path/module_a/getpath.py
os.path.split(os.path.realpath(__file__))[0] is /Users/vk/tmp/path/module_a
---------------------------------
------ module_a/module_b/getpath.py ------
os.getcwd() is /Users/vk/tmp/path
sys.path[0] is /Users/vk/tmp/path/module_a
sys.argv[0] is module_a/getpath.py
__file__ is /Users/vk/tmp/path/module_a/module_b/getpath.py
os.path.dirname(__file__) is /Users/vk/tmp/path/module_a/module_b
os.path.abspath(__file__) is /Users/vk/tmp/path/module_a/module_b/getpath.py
os.path.realpath(__file__) is /Users/vk/tmp/path/module_a/module_b/getpath.py
os.path.split(os.path.realpath(__file__))[0] is /Users/vk/tmp/path/module_a/module_b
------------------------------------------

由实验结果,可以得出如下结论。

获取执行命令的绝对路径

1
os.getcwd()

获取入口脚本的父绝对路径

1
sys.path[0]

获取执行脚本的绝对路劲

1
2
os.path.abspath(__file__)
os.path.realpath(__file__)

获取执行脚本的父绝对路径

1
os.path.split(os.path.realpath(__file__))[0]

打印日志

输出日志到控制台

1
2
3
4
5
6
7
8
import sys
import logging

log = logging.getLogger(__name__)
log.setLevel(logging.DEBUG)
log.addHandler(logging.StreamHandler(sys.stdout)) #默认sys.error

log.info('print info level log to console')

输出日志到文件

1
2
3
4
5
6
7
8
import sys
import logging

log = logging.getLogger(__name__)
log.setLevel(logging.DEBUG)
log.addHandler(logging.FileHandler('python.log'))

log.info('print info level log to file')

日志文件默认存储到执行命令的路径下,可以通过使用绝对路径来指定日志文件路径。

封装日志模块

1、封装一个日志模块 log.py

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
# -*- coding: utf-8 -*-

import os
import logging
import logging.handlers

def init_log(log_path, level=logging.INFO, when="D", backup=7,
format="%(levelname)s: %(asctime)s: %(filename)s:%(lineno)d * %(thread)d %(message)s",
datefmt="%m-%d %H:%M:%S"):
"""
init_log - initialize log module

Args:
log_path - Log file path prefix.
Log data will go to two files: log_path.log and log_path.log.wf
Any non-exist parent directories will be created automatically
level - msg above the level will be displayed
DEBUG < INFO < WARNING < ERROR < CRITICAL
the default value is logging.INFO
when - how to split the log file by time interval
'S' : Seconds
'M' : Minutes
'H' : Hours
'D' : Days
'W' : Week day
default value: 'D'
format - format of the log
default format:
%(levelname)s: %(asctime)s: %(filename)s:%(lineno)d * %(thread)d %(message)s
INFO: 12-09 18:02:42: log.py:40 * 139814749787872 HELLO WORLD
backup - how many backup file to keep
default value: 7

Raises:
OSError: fail to create log directories
IOError: fail to open log file
"""
formatter = logging.Formatter(format, datefmt)
logger = logging.getLogger()
logger.setLevel(level)

dir = os.path.dirname(log_path)
if not os.path.isdir(dir):
os.makedirs(dir)

handler = logging.handlers.TimedRotatingFileHandler(log_path + ".log",
when=when,
backupCount=backup)
handler.setLevel(level)
handler.setFormatter(formatter)
logger.addHandler(handler)

handler = logging.handlers.TimedRotatingFileHandler(log_path + ".log.wf",
when=when,
backupCount=backup)
handler.setLevel(logging.WARNING)
handler.setFormatter(formatter)
logger.addHandler(handler)

2、使用日志模块

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import logging

import log

class VKTest(object):
def __init__(self):
logging.info('print info level log to file')

if __name__=="__main__":
# 日志保存到./log/vk.log和./log/vk.log.wf,按天切割,保留7天
log.init_log("./log/vk")
logging.info("程序启动!!!")

wiki_export = VKTest()

logging.info("程序结束!!!")

参考文档

  • 本文作者: 好好学习的郝
  • 本文链接: https://www.voidking.com/dev-python-tricks/
  • 版权声明: 本博客所有文章除特别声明外,均采用 BY-NC-SA 许可协议。转载请注明出处!源站会及时更新知识点及修正错误,阅读体验也更好。欢迎分享,欢迎收藏~