网站更新内容:请访问: https://bigdata.ministep.cn/

python 命令行工具

有时候,需要借助命令行工具进行交互操作

这里提供一个范例

## ball.py 文件名
import requests
import time 
import pysnowball as ball
import argparse

def get_token():
  r = requests.get("https://xueqiu.com", headers={"user-agent":"Mozilla"})
  token = r.cookies["xq_a_token"]
  # print(r.cookies["xq_a_token"])
  return token

def cmd():
  parser = argparse.ArgumentParser(description='Process some integers.')
  parser.add_argument('-s','--stock',default='SH000300',type=str,dest = "stock", help='an interger for the ball')
  args = parser.parse_args()#返回一个命名空间,如果想要使用变量,可用args.attr
  print(args)
  snow_code = args.stock
  return snow_code

if __name__ == "__main__":
    start_time = time.time() # 开始时间
    token_list = []
    token = get_token()
    token_list.append(token)
    ball.set_token('xq_a_token={token};'.format(token=token_list[0]))
    snow_code = cmd()
    # print(snow_code)
    # snow_code = ball.cash_flow('SH600000')
    snow_current=ball.quotec(snow_code)
    print(snow_current)
    end_time = time.time() #结束时间
    # print("程序耗时%f秒." % (end_time - start_time))

在cmd交互运行:
python ball.py -s SH600486

难点:argparse 参数的解析使用
可参考:argparse简要用法总结 | Yunfeng's Simple Blog]

不过要尤其注意
parser.add_argument('-s','--stock',default='SH000300',type=str,dest = "stock", help='an interger for the ball')

args.stock参数的对应关系

posted @ 2022-02-16 15:15  ministep88  阅读(333)  评论(0)    收藏  举报
网站更新内容:请访问:https://bigdata.ministep.cn/