use with to open and close a file

zhaoqifa · zhaoqifa · commit d4eccca9fbfe · 2015-11-26T12:43:01.000+08:00
diff --git a/python/hangover.py b/python/hangover.py
@@ -42,16 +42,9 @@
 )
 
 try:
-    f = open('logs/file.txt', 'a')
-except IOError as e:
-    # dir & file don't exist; create them
-    os.mkdir('logs')
-    f = open('logs/file.txt', 'a')
+    if not os.path.exists('logs'):
+        os.mkdir('logs')
+    with open('logs/file.txt', 'a') as lh:
+        lh.write("Message sent at " + strftime("%a, %d %b %Y %H:%M:%S") + "\n")
 except Exception as e:
     print e
-else:
-    pass
-
-# log it
-f.write("Message sent at " + strftime("%a, %d %b %Y %H:%M:%S") + "\n")
-f.close()
diff --git a/python/smack_my_bitch_up.py b/python/smack_my_bitch_up.py
@@ -43,16 +43,9 @@
 )
 
 try:
-    f = open('logs/file.txt', 'a')
-except IOError as e:
-    # dir & file don't exist; create them
-    os.mkdir('logs')
-    f = open('logs/file.txt', 'a')
+    if not os.path.exists('logs'):
+        os.mkdir('logs')
+    with open('logs/file.txt', 'a') as lh:
+        lh.write("Message sent at " + strftime("%a, %d %b %Y %H:%M:%S") + "\n")
 except Exception as e:
     print e
-else:
-    pass
-
-# log it
-f.write("Message sent at " + strftime("%a, %d %b %Y %H:%M:%S") + "\n")
-f.close()
