检测文件是否存在并读取内容

tuntun · tuntun · commit dcae846fd407 · 2015-11-05T17:38:44.000+08:00
diff --git a/checkfile.py b/checkfile.py
@@ -0,0 +1,30 @@
+import sys		# Import the Modules
+import os		# Import the Modules
+
+# Readfile Functions which open the file that is passed to the script
+
+def readfile(filename):
+    print(filename)
+    f = open(filename, 'r')
+    line = f.read()
+    print(line)
+
+
+def main():
+    filename = ''
+    if len(sys.argv) == 2:		# Check the arguments passed to the script
+        filename = sys.argv[1]		# The filename is the first argument
+        if not os.path.isfile(filename):	# Check the File exists
+            print('[-] ' + filename + ' does not exist.')
+            exit(0)
+        if not os.access(filename, os.R_OK):	# Check you can read the file
+            print('[-] ' + filename + ' access denied')
+            exit(0)
+    else:
+        print('[-] Usage: ' + str(sys.argv[0]) + ' <filename>')  # Print usage if not all parameters passed/Checked
+        exit(0)
+    print('[+] Reading from : ' + filename)  # Display Message and read the file contents
+    readfile(filename)
+  
+if __name__ == '__main__':
+  main()
