《求教》修改一个脚本

skyeydemon

写了这么一个脚本：

#!/usr/bin/env python
###############################################
#List the date you specified the range of time            
###############################################
import datetime

TheStartDay = datetime.date(2010,8,15)
TheEndofDay = datetime.date(2010,9,1)

TheStart = TheStartDay.strftime("%Y%m%d")
TheEnd = TheEndofDay.strftime("%Y%m%d")

while TheStart <= TheEnd:
        print TheStart,
        TheStartDay = TheStartDay + datetime.timedelta(days=1)
        TheStart = TheStartDay.strftime("%Y%m%d")

执行：
# ./time_range.py
20100815 20100816 20100817 20100818 20100819 20100820 20100821 20100822 20100823 20100824 20100825 20100826 20100827 20100828 20100829 20100830 20100831 20100901


是用来实现列出所指定起始日期和终止日期之间的所有时间的。现在功能已经达到了，但是有点问题。
想把脚本改成如下方式：

# ./time_range.py  20100815 20100901
就能输出，不用修改里面的内容。
这个要怎么做？
用sys.argv实现么？

chinaunix874

用了个笨办法，楼主看看能满足需求不。

1. 
   
2. #!/usr/bin/env python
   
3. ###############################################
   
4. #List the date you specified the range of time            
   
5. ###############################################
   
6. import datetime
   
7. import sys
   
8. 
   
9. def usage(e):
   
10. 
    
11.     print """Usage: time_range.py [start_time] [end_time]
    
12.                     start_time: YYYYMMDD    end_time: YYYYMMDD
    
13.           """
    
14.     sys.exit(e)
    
15. 
    
16. 
    
17. #TheStartDay = datetime.date(2010,8,15)
    
18. #TheEndofDay = datetime.date(2010,9,1)
    
19. if len(sys.argv) != 3:
    
20.     usage(1)
    
21. 
    
22. 
    
23. start = sys.argv[1]
    
24. end = sys.argv[2]
    
25. 
    
26. 
    
27. if (len(start) != 8) or (len(end) != 8):
    
28.     usage(1)
    
29. 
    
30. start_year = int(start[0:4])
    
31. start_month = int(start[4:6])
    
32. start_day = int(start[6:])
    
33. 
    
34. end_year = int(end[0:4])
    
35. end_month = int(end[4:6])
    
36. end_day = int(end[6:])
    
37. 
    
38. 
    
39. TheStartDay = datetime.date(start_year, start_month, start_day)
    
40. TheEndofDay = datetime.date(end_year, end_month, end_day)
    
41. 
    
42. TheStart = TheStartDay.strftime("%Y%m%d")
    
43. TheEnd = TheEndofDay.strftime("%Y%m%d")
    
44. 
    
45. while TheStart <= TheEnd:
    
46.         print TheStart,
    
47.         TheStartDay = TheStartDay + datetime.timedelta(days=1)
    
48.         TheStart = TheStartDay.strftime("%Y%m%d")
    

复制代码

提取时间参数的办法比较弱智，抛砖引玉吧。

skyeydemon

回复 2# chinaunix874


    呵呵，多谢了。功能上已经是满足了。
学习了。。。。