几个python入门脚本

database=[
["zhang3","0111"],
["li4","0112"],
["wang5","0113"]
]
username=raw_input("what's your user name?")
id=raw_input("what's your id?")
if [username,id] in database: print "access granted"

def bj(x):
    if x<0:
         print "the number is too small"
    elif x>100:
         print "the number is too big"
    else: print "the number is right!"

x=input("please input a number:")
bj(x)

读取文件的脚本

import os
if os.getcwd()!='/root':
    os.chdir('/root')

try:
    with open('hi.txt','r') as f:
        print(f.readline(), end='')
    f.close()
except IOError as err:    #此写法仅适用于python3
    print('File Error:' + str(err))

#!/usr/bin/python
import os
path1=os.getcwd()
path2=os.path.abspath('/usr/local/5.txt')
print('使用os.getcwd()获取的内容为' + path1)
print('使用os.path.abspath()获取的内容为' + path2)

统计文件夹大小的脚本

import os  
from os.path import join, getsize  
  
def getdirsize(dir):  
   size = 0  
   for root, dirs, files in os.walk(dir):  
      size += sum([getsize(join(root, name)) for name in files])  
   return size  
  
if __name__ == '__main__':
   filesize = getdirsize('/tmp')  
   print ('There are ' + str(filesize/1024/1024) + 'Mbytes in /tmp')

文件写入的简单脚本

import zipfile
f = zipfile.ZipFile('archive.zip','w',zipfile.ZIP_DEFLATED)
f.write('1.py')
f.write('/root/install.log')
f.close()

压缩文件夹的脚本

#!/usr/bin/python
import zipfile
import os
f = zipfile.ZipFile('archive.zip','w',zipfile.ZIP_DEFLATED)
startdir = "/root/mydir"
for dirpath, dirnames, filenames in os.walk(startdir):
    for filename in filenames:
        f.write(os.path.join(dirpath,filename))
f.close()

调用shell命令的脚本

#!/usr/bin/python
# -*- coding: utf-8 -*-
import os
import commands
a=commands.getstatusoutput('pwd')
b=commands.getstatusoutput('pwdaa')
print('正确命令的返回状态是: ' + str(a[0]))
print('正确命令的返回结果是: ' + a[1])
print('错误命令的返回状态是: ' + str(b[0]))
print('错误命令的返回结果是: ' + b[1])
c=os.popen('pwd')
print c.read().rstrip()