pythoncmd模块_Python使用CMD模块更优雅的运行脚本

本文实例讲述了Python使用CMD模块更优雅的运行脚本的方法。分享给大家供大家参考。具体分析如下&＃xff1a;

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

python的cmd模块提供的更优雅的方式&＃xff0c;很喜欢。

刚写了一些平时常用的测试代码&＃xff1a;

# -*- 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 &＃61; func(*args, **kwargs)

return res

except:

print "there is a wrong in %s, please input &＃39;help %s&＃39; 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 &＃61; &＃39;> &＃39;

#配置django环境

pathname &＃61; 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), &＃39;../&＃39;)))

os.environ[&＃39;DJANGO_SETTINGS_MODULE&＃39;] &＃61; &＃39;settings&＃39;

#清用户cache

def help_clear_cache(self):

print "Usage:clear_cache [sns_id]"

print "-- clear the user&＃39;s cache"

&＃64;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"

&＃64;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 &＃61; parms.split()

sns_id &＃61; int(sns_id)

level &＃61; int(level)

member &＃61; get_member_by_sns_id(sns_id)

al_info &＃61; get_or_create_almanac_info(member.id)

al_info.exp &＃61; EXP_LEVEL[level-1] &＃43; 1

al_info.save()

#增加用户珍珠

def help_add_pearls(self):

print "Usage:add_pearls [sns_id] [pearls]"

print "-- add some pearls to the user"

&＃64;not_exit

def do_add_pearls(self, parms):

from platforms.account.helper import add_account_pearls

sns_id, pearls &＃61; parms.split()

sns_id &＃61; int(sns_id)

pearls &＃61; int(pearls)

add_account_pearls(&＃39;&＃39;, &＃39;&＃39;, sns_id, pearls, type&＃61;&＃39;Test&＃39;, money&＃61;0)

#分鱼

def help_add_fishes(self):

print "Usage:add_fishes [sns_id] [fish1 fish2 ...]"

print "-- add some fishes to the user"

&＃64;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 &＃61; parms.split()

sns_id &＃61; int(array[0])

fish_ids &＃61; [int(i) for i in array[1:]]

member &＃61; get_member_by_sns_id(sns_id)

for fish_id in fish_ids:

fish &＃61; get_fish(fish_id)

new_a_member_fish(member, fish.id, update_almanac&＃61;True)

#分贝壳

def help_add_shells(self):

print "Usage:add_shells [sns_id] [shells]"

print "-- add some shells to the user"

&＃64;not_exit

def do_add_shells(self, parms):

from apps.members.cache import get_member_by_sns_id

sns_id, num &＃61; parms.split()

sns_id &＃61; int(sns_id)

num &＃61; int(num)

member &＃61; get_member_by_sns_id(sns_id)

member.shells &＃43;&＃61; 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"

&＃64;not_exit

def do_add_fish_food(self, parms):

from apps.members.cache import get_member_by_sns_id

sns_id, num &＃61; parms.split()

sns_id &＃61; int(sns_id)

num &＃61; int(num)

member &＃61; get_member_by_sns_id(sns_id)

member.fish_food &＃43;&＃61; 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"

&＃64;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 &＃61; parms.split()

sns_id &＃61; int(array[0])

tank_ids &＃61; [int(i) for i in array[1:]]

member &＃61; get_member_by_sns_id(sns_id)

for tank_id in tank_ids:

tank &＃61; 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"

&＃64;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 &＃61; parms.split()

sns_id &＃61; int(array[0])

item_ids &＃61; [int(i) for i in array[1:]]

member &＃61; get_member_by_sns_id(sns_id)

for item_id in item_ids:

item &＃61; 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&＃39;s task"

&＃64;not_exit

def do_refresh_task(self, sns_id):

from apps.members.cache import get_member_by_sns_id

sns_id &＃61; int(sns_id)

member &＃61; get_member_by_sns_id(sns_id&＃61;sns_id)

member.task_start_time &＃61; None

member.save()

#调合成等级

def help_synthesis_level_up(self):

print "Usage:synthesis_level_up [sns_id] [level]"

print "-- level up user&＃39;s synthesis level"

&＃64;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 &＃61; parms.split()

sns_id &＃61; int(sns_id)

lev &＃61; int(lev)

member &＃61; get_member_by_sns_id(sns_id&＃61;sns_id)

exp &＃61; int(sum([i**1.5*20 for i in range(1,lev&＃43;1)]))

member &＃61; get_member_by_sns_id(sns_id)

synth &＃61; get_or_create_synth_info(member.id)

synth.synth_exp &＃61; exp

synth.save()

def help_quit(self):

print "Usage:quit"

def do_quit(self, line):

exit()

do_q &＃61; do_quit

if __name__ &＃61;&＃61; "__main__":

nb_script &＃61; NBScript()

nb_script.cmdloop()

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

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

本文原创发布php中文网&＃xff0c;转载请注明出处&＃xff0c;感谢您的尊重&＃xff01;