python使用cmd模块更优雅的运行脚本

本文实例讲述了python使用cmd模块更优雅的运行脚本的方法。分享给大家供大家参考。具体分析如下:

平时由于经常给测试人员调试一些东西,虽然写了一些脚本,感觉还是不方便。

python的cmd模块提供的更优雅的方式,很喜欢。

刚写了一些平时常用的测试代码:

# -*- coding:utf-8 -*-
import os, sys
from datetime import datetime
import cmd
import warnings
from django.conf import settings
def not_exit(func):
def _func(*args, **kwargs):
try:
res = func(*args, **kwargs)
return res
except:
print “there is a wrong in %s, please input ‘help %s’ to get help” % (func.__name__, func.__name__[3:])
return none
return _func
class nbscript(cmd.cmd):
def __init__(self):
cmd.cmd.__init__(self)
#设置命令提示符
self.prompt = ‘> ‘
#配置django环境
pathname = os.path.dirname(sys.argv[0])
sys.path.append(os.path.abspath(pathname))
sys.path.append(os.path.normpath(os.path.join(os.path.abspath(pathname), ‘../’)))
os.environ[‘django_settings_module’] = ‘settings’
#清用户cache
def help_clear_cache(self):
print “usage:clear_cache [sns_id]”
print “– clear the user’s cache”
@not_exit
def do_clear_cache(self, sns_id):
from scripts.update_cache import clear_member_all_cache
clear_member_all_cache(sns_id)
#用户升级
def help_user_level_up(self):
print “usage:user_level_up [sns_id] [level]”
print “– level up the user to the level”
@not_exit
def do_user_level_up(self, parms):
from apps.members.cache import get_member_by_sns_id, get_or_create_almanac_info
from apps.members.constants import exp_level
sns_id, level = parms.split()
sns_id = int(sns_id)
level = int(level)
member = get_member_by_sns_id(sns_id)
al_info = get_or_create_almanac_info(member.id)
al_info.exp = exp_level[level-1] + 1
al_info.save()
#增加用户珍珠
def help_add_pearls(self):
print “usage:add_pearls [sns_id] [pearls]”
print “– add some pearls to the user”
@not_exit
def do_add_pearls(self, parms):
from platforms.account.helper import add_account_pearls
sns_id, pearls = parms.split()
sns_id = int(sns_id)
pearls = int(pearls)
add_account_pearls(”, ”, sns_id, pearls, type=’test’, money=0)
#分鱼
def help_add_fishes(self):
print “usage:add_fishes [sns_id] [fish1 fish2 …]”
print “– add some fishes to the user”
@not_exit
def do_add_fishes(self, parms):
from apps.members.cache import get_member_by_sns_id
from apps.members.helper import new_a_member_fish
from apps.products.cache import get_fish
array = parms.split()
sns_id = int(array[0])
fish_ids = [int(i) for i in array[1:]]
member = get_member_by_sns_id(sns_id)
for fish_id in fish_ids:
fish = get_fish(fish_id)
new_a_member_fish(member, fish.id, update_almanac=true)
#分贝壳
def help_add_shells(self):
print “usage:add_shells [sns_id] [shells]”
print “– add some shells to the user”
@not_exit
def do_add_shells(self, parms):
from apps.members.cache import get_member_by_sns_id
sns_id, num = parms.split()
sns_id = int(sns_id)
num = int(num)
member = get_member_by_sns_id(sns_id)
member.shells += num
member.save()
#分鱼食
def help_add_fish_food(self):
print “usage:add_fish_food [sns_id] [food_num]”
print “– add some food to the user”
@not_exit
def do_add_fish_food(self, parms):
from apps.members.cache import get_member_by_sns_id
sns_id, num = parms.split()
sns_id = int(sns_id)
num = int(num)
member = get_member_by_sns_id(sns_id)
member.fish_food += num
member.save()
#分鱼缸
def help_add_fish_tanks(self):
print “usage:add_fish_food [sns_id] [tank1,tank2 …]”
print “– add some tanks to the user”
@not_exit
def do_add_fish_tanks(self, parms):
from apps.members.cache import get_member_by_sns_id
from apps.members.helper import new_a_member_tank
from apps.products.cache import get_fish_tank
array = parms.split()
sns_id = int(array[0])
tank_ids = [int(i) for i in array[1:]]
member = get_member_by_sns_id(sns_id)
for tank_id in tank_ids:
tank = get_fish_tank(tank_id)
new_a_member_tank(member, tank.id)
#分道具
def help_add_items(self):
print “usage:add_items [sns_id] [itme1,item2 …]”
print “– add some itmes to the user”
@not_exit
def do_add_items(self, parms):
from apps.members.cache import get_member_by_sns_id
from apps.members.helper import new_a_member_item
from apps.products.cache import get_item
array = parms.split()
sns_id = int(array[0])
item_ids = [int(i) for i in array[1:]]
member = get_member_by_sns_id(sns_id)
for item_id in item_ids:
item = get_item(item_id)
new_a_member_item(member, item.id)
#刷任务
def help_refresh_task(self):
print “usage:refresh_task [sns_id]”
print “– refresh user’s task”
@not_exit
def do_refresh_task(self, sns_id):
from apps.members.cache import get_member_by_sns_id
sns_id = int(sns_id)
member = get_member_by_sns_id(sns_id=sns_id)
member.task_start_time = none
member.save()
#调合成等级
def help_synthesis_level_up(self):
print “usage:synthesis_level_up [sns_id] [level]”
print “– level up user’s synthesis level”
@not_exit
def do_synthesis_level_up(self, parms):
from apps.members.cache import get_member_by_sns_id
from apps.synthesis.cache import get_or_create_synth_info
sns_id, lev = parms.split()
sns_id = int(sns_id)
lev = int(lev)
member = get_member_by_sns_id(sns_id=sns_id)
exp = int(sum([i**1.5*20 for i in range(1,lev+1)]))
member = get_member_by_sns_id(sns_id)
synth = get_or_create_synth_info(member.id)
synth.synth_exp = exp
synth.save()
def help_quit(self):
print “usage:quit”
def do_quit(self, line):
exit()
do_q = do_quit
if __name__ == “__main__”:
nb_script = nbscript()
nb_script.cmdloop()

这里权当cmd模块的简单学习了。

希望本文所述对大家的python程序设计有所帮助。

Posted in 未分类