코드 다시 수정

+߸   

-#!/usr/bin/env python
-# coding: utf-8
-
-# In[ ]:
-
-
 import GetOldTweets3 as got
 from bs4 import BeautifulSoup
 
 import datetime
-
-days_range = []
-
-start = datetime.datetime.strptime("2019-11-14", "%Y-%m-%d")
-end = datetime.datetime.strptime("2019-11-15", "%Y-%m-%d")
-date_generated = [start + datetime.timedelta(days=x) for x in range(0, (end-start).days)]
-
-for date in date_generated:
-    days_range.append(date.strftime("%Y-%m-%d"))
-print("=== 설정된 트윗 수집 기간은 {} 에서 {} 까지 입니다 ===".format(days_range[0], days_range[-1]))
-print("=== 총 {}일 간의 데이터 수집 ===".format(len(days_range)))
-
 import time
-
-# 수집 기간 맞추기
-start_date = days_range[0]
-end_date = (datetime.datetime.strptime(days_range[-1], "%Y-%m-%d") 
-            + datetime.timedelta(days=1)).strftime("%Y-%m-%d") # setUntil이 끝을 포함하지 않으므로, day + 1
-
-# 트윗 수집 기준 정의
-tweetCriteria = got.manager.TweetCriteria().setQuerySearch('한글')                                           .setSince(start_date)                                           .setUntil(end_date)                                           .setMaxTweets(-1)
-
-# 수집 with GetOldTweet3
-print("Collecting data start.. from {} to {}".format(days_range[0], days_range[-1]))
-start_time = time.time()
-
-tweet = got.manager.TweetManager.getTweets(tweetCriteria)
-
-print("Collecting data end.. {0:0.2f} Minutes".format((time.time() - start_time)/60))
-print("=== Total num of tweets is {} ===".format(len(tweet)))
 from random import uniform
 from tqdm import tqdm_notebook
 
-# initialize
-tweet_list = []
+def get_tweets(criteria):
+    tweet = got.manager.TweetManager.getTweets(criteria)
+    tweet_list = []
 
-for index in tqdm_notebook(tweet):
+    for index in tqdm_notebook(tweet):
     
         # 메타데이터 목록 
         username = index.username
@@ -61,9 +27,82 @@ for index in tqdm_notebook(tweet):
         # 휴식 
         time.sleep(uniform(1,2))
 
+days_range = []
 
-# In[ ]:
-
+start = datetime.datetime.strptime("2019-11-17", "%Y-%m-%d")
+end = datetime.datetime.strptime("2019-11-18", "%Y-%m-%d")
+date_generated = [start + datetime.timedelta(days=x) for x in range(0, (end-start).days)]
 
+for date in date_generated:
+    days_range.append(date.strftime("%Y-%m-%d"))
+print("=== 기본으로 설정된 트윗 수집 기간은 {} 에서 {} 까지 입니다 ===".format(days_range[0], days_range[-1]))
+print("=== 총 {}일 간의 데이터 수집 ===".format(len(days_range)))
 
+# 수집 기간 맞추기
+start_date = days_range[0]
+end_date = (datetime.datetime.strptime(days_range[-1], "%Y-%m-%d") 
+            + datetime.timedelta(days=1)).strftime("%Y-%m-%d") # setUntil이 끝을 포함하지 않으므로, day + 1
 
+my_key = input("검색할 키워드를 입력해주세요: ")
+
+while(True):
+    temp1 = "현재 검색어는 " + my_key + "입니다. "
+    print(temp1)
+    print("기간은 기본적으로 최근 1일입니다.")
+    print("1. 닉네임을 통한 검색")
+    print("2. 키워드를 통한 검색")
+    print("3. 시간을 통한 검색")
+    print("4. 종료")
+    userNum = int(input("무엇을 하시겠습니까?: "))
+    
+    if userNum == 1:
+        nick = input("검색할 닉네임을 입력해주세요: ")
+        tweetCriteria = got.manager.TweetCriteria().setUsername(nick)\
+                                           .setQuerySearch(my_key)\
+                                           .setSince(start_date)\
+                                           .setUntil(end_date)\
+                                           .setMaxTweets(-1)
+        get_tweets(tweetCriteria)
+    elif userNum == 2:
+        my_key = input("검색할 키워드를 입력해주세요: ")
+        tweetCriteria = got.manager.TweetCriteria().setQuerySearch(my_key)\
+                                           .setSince(start_date)\
+                                           .setUntil(end_date)\
+                                           .setMaxTweets(-1)
+        get_tweets(tweetCriteria)
+    elif userNum == 3:
+        user_start = int(input("시작일을 입력해주세요(yyyymmdd형태): "))
+        if(user_start < 20170000 or user_start > 20191200):
+            print("최근 3년 이내만 검색가능합니다.")
+            continue
+        user_end = int(input("종료일을 입력해주세요(yyyymmdd형태): "))
+        if(user_end > 20191200):
+            print("미래로 갈 수는 없습니다.")
+            continue
+        elif(user_end < user_start):
+            print("시작일보다 작을 수 없습니다.")
+            continue
+        if(user_end - 8 > user_start):
+            print("최대 1주일까지 검색이 가능합니다.")
+            continue
+        else:
+            start_year = user_start // 10000
+            start_month = user_start // 100 - start_year * 100
+            start_day = user_start - start_year * 10000 - start_month * 100
+            end_year = user_end // 10000
+            end_month = user_end // 100 - end_year * 100
+            end_day = user_end - end_year * 10000 - end_month * 100
+            d1 = str(start_year) + "-" + str(start_month) + "-" + str(start_day)
+            d2 = str(end_year) + "-" + str(end_month) + "-" + str(end_day)
+            print("=== 현재 설정된 트윗 수집 기간은 {} 에서 {} 까지 입니다 ===".format(d1, d2))
+            print("=== 총 {}일 간의 데이터 수집 ===".format(user_end - user_start))
+            tweetCriteria = got.manager.TweetCriteria().setQuerySearch(my_key)\
+                                           .setSince(d1)\
+                                           .setUntil(d2)\
+                                           .setMaxTweets(-1)
+            get_tweets(tweetCriteria)
+    elif userNum == 4:
+        break
+    else:
+        print("잘못된 입력입니다.")
+        continue