好好学习，天天向上！

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 许可协议。转载请注明出处！源站会及时更新知识点及修正错误，阅读体验也更好。欢迎分享，欢迎收藏~