使用 python 开发了一个能在命令行中查看基金涨幅的摸鱼工具

1,241 阅读4分钟

fundgz 命令行工具

一朋友在上班期间看手机基金信息,但上班期间频繁看手机着实不太好,于是就想到写一个能在命令行查看 fund 估值的命令行工具,通过输入 fund 编号和名称,就能查询到当前时间的估值。 网络上有很多这样的摸鱼插件,甚至在 vscode 里的插件比比皆是。 本着练手的态度,自己也写了一个简单的工具,就当做学习新的库了。

使用预览

安装工具:

pip install fundgz

使用:

> fundgz

Usage: fundgz [OPTIONS] COMMAND [ARGS]...

Options:
  --help  Show this message and exit.

Commands:
  add
  delete
  run

命令行输入 fundgz 命令,可以查看这个命令的简介,可以看到提供了三个子命令:

> fundgz add

正在添加 fund 文档,ctrl+d 退出
请输入fund 编号: 078943
请输入fund 名称: test

输入 fundgz add 命令,终端会提示输入 fund 编号和名称,输入完成后,可以按 ctrl+d 退出。

fundgz delete 命令删除录入文档的编号

fundgz run 命令就是根据之前存入的文档执行查询了:

在终端会展示结果表格

fund.png

开发思路

  • 在终端录入 fund code 并保存至临时文件中
  • 查询临时文件,提取其中的 fund code 列表
  • 通过 code 列表,批量查询 fund 当日的估值信息
  • 在终端列表展示

技术栈

python3.7 版本

数据抓取 - asyncio + aiohttp

命令行工具 - click + inquirer

终端展示 - rich

打包发布 - setuptools

click + inquirer 终端工具

使用 click.group() 创建命令组,实现 add, deleterun 命令

@click.group()
def fc():
    pass

@fc.command()
def run():
    pass

@fc.command()
def add():
    pass
    
@fc.command()
def delete():
    pass
    
fc()

@fc.command() 是为了替代 fc.add_command(run) ,这样写更加优雅

@click.command()
def run():
    pass

fc.add_command(run)

inquirer 库是交互式工具,提供了很多便利的操作,这次使用的是 list 选择。

执行 fundgz delete 选择要删除的 code

fund-del.png

del_list = [
                inquirer.List('del',
                              message="请选择要删除的 fund:",
                              choices=codes)
            ]
            del_answers = inquirer.prompt(del_list)

asyncio + aiohttp 批量数据查询

读取临时文件后,提取出正在需要的 code 列表,通过这个列表批量查询数据

async with aiohttp.ClientSession() as session:
    tasks = [asyncio.create_task(query_data(session, code)) for code in codes]
    data = await asyncio.gather(*tasks)
    print(data)

如果没有查询到数据,则需要做好异常处理,则请求非 200 的则直接返回 None:

async def query_data(session, fund_code):
    """
    :param fund_code:
    :return:
    """
    url = BASE_URL.format(fund_code)
    async with session.get(url=url) as response:
        if response.status == 200:
            bytes_data = await response.read()
            # 查看编码类型
            ret = chardet.detect(bytes_data)
            # bytes 转换成 str
            jsonp_data = str(bytes_data, ret['encoding'])
            # 判断是否有数据
            data = re.match('.*?({.*}).*', jsonp_data, re.S).group(1)
            return json.loads(data)
        else:
            return None

如此,批量返回的数据中就会有 None,所以要 filter

data = filter(lambda fund: fund, data)

setuptools 打包

命令行工具已经可以了,使用

python fundgz/main.py run

但是期望的是

fundgz run

所以,就要将这个工具发布到服务上。

首先在 https://pypi.org/ 上注册一个账号,认证邮箱等

其次,安装 setuptools,更新至最新版:

python -m pip install --upgrade pip setuptools wheel

具体可以查看官网 https://pypi.org/project/setuptools/

接下来就是打包的事儿了。

在项目根目录下新建两个文件 setup.pysetup.cfg

此时项目的结构是

├── README.md
├── fund-code.txt
├── fundgz
│   ├── __init__.py
│   └── main.py
├── setup.cfg
└── setup.py

setup.py 文件内,填写打包的必要信息:

#!usr/bin/env python
# -*- coding:utf-8 _*-

from setuptools import setup, find_packages

VERSION = '0.1.7'

setup(
    name='fundgz',
    version=VERSION,
    description='fund search',
    long_description='fund search',
    keywords='fund asynico click rich inquirer',
    author='caoshiping',
    author_email='soraping@163.com',
    url='https://github.com/soraping/fc',
    license='MIT',
    packages=find_packages(exclude=["*.txt"]),
    include_package_data=True,
    zip_safe=True,
    install_requires=[
        'aiohttp',
        'inquirer',
        'click',
        'rich'
    ],
    python_requires='>=3.7',
    entry_points={
        'console_scripts': [
            'fundgz = fundgz.main:main'
        ]
    }
)

关键字段详解:

  • name

上传包的名称

  • version

开发库版本

  • packages

打包的内容目录,使用 find_packages 方法,使用参数 exclude 屏蔽一些文件

  • install_requires

所依赖的库列表

  • entry_points

自定义命令,不用 python xxxx.py 这样执行的,就要配置这个字段。这个字段不仅仅这个功能,当前包通过 setuptools 注册的一个外部可以直接调用的接口。

    entry_points={
        'console_scripts': [
            'fundgz = fundgz.main:main'
        ]
    }

在终端输入:

$ fundgz --help
usage: fundgz...

console_scripts 是自定义命令行列表

fundgz = fundgz.main:mainfundgz 是命令名称,等号右边的是主方法路径,其中 fundgz.main 包根目录下执行文件的路径,最后是执行文件的方法名

setup.py 文件配置完成后就可以执行打包操作了

  1. 在终端执行命令:
# 使用sdist构建源码分发包
python setup.py sdist bdist_wheel

这里是固定的命令,且在 setup.py 文件相同的目录下执行,当这个命令运行结束后,确保在生成的 dist/ 文件夹下存在相应的 .whl 文件和 .tar.gz 文件。其中 .tar.gz 文件是我们 的python package 的源文件文档。

.whl 是一个软件分发包 (build distribution) 。新版本的 pip 将会首先尝试安装软件分发包,但在失败情况下会接着尝试采用源文件包安装。

执行打包后,打开 dist 文件夹,可以发现有 setup.py 配置的 nameversion 拼接的包文件,这些就是待上传的包

  1. 上传包

需要安装上传工具 twine

pip install --user --upgrade twine

# 上传
twine upload dist/*

上传的时候,会提示输入 pypi 账号密码。

也可以指定版本号上传

twine upload dist/fundgz-0.1.7*

到此,一个包就算上传成功了,可以下载下来试下

> pip install fundgz --upgrade

> fundgz --help

项目地址

https://github.com/soraping/fc

参考文章

  • click

blog.csdn.net/weixin_3827…

www.cnblogs.com/alexkn/p/69…

  • setuptools

www.cnblogs.com/Zzbj/p/1153…

blog.csdn.net/u011519550/…

blog.csdn.net/yinshuilan/…

www.jb51.net/article/178…

www.jianshu.com/p/eb27d5cb5…