haha

currenjin · currenjin · commit 6d248891f034 · 2020-08-27T00:27:16.000+09:00
diff --git a/application/restaurant.csv b/application/restaurant.csv
@@ -1,2 +1 @@
-Name,Count
-Apple,1
+
diff --git a/application/restaurant_bot.py b/application/restaurant_bot.py
@@ -21,10 +21,14 @@
     csv_file.seek(0)
     for row in reader:
         if restaurant in row['Name']:
-            row['Count'] = 2
+            with open('restaurant.csv', 'w'):
+                row['Count'] = 2
             break
         else:
             writer.writerow({'Name': restaurant, 'Count': 1})
             break
+    else:
+        writer.writeheader()
+        writer.writerow({'Name': restaurant, 'Count': 1})
 
 print('감사합니다. {}님, 좋은 하루 보내세요!'.format(name))
