作者:陈先森的记忆 | 来源:互联网 | 2023-09-07 16:50
有时候,需要借助命令行工具进行交互操作这里提供一个范例##ball.py文件名importrequestsimporttimeimportpysnowballasballimpor
有时候,需要借助命令行工具进行交互操作
这里提供一个范例
## 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
参数的对应关系