store id_list

PeiShang · Apr 12, 2018 · 2e6bb0c · 2e6bb0c
1 parent 710780d
commit 2e6bb0c
Show file tree

Hide file tree

Showing 6 changed files with 70 additions and 48 deletions.
diff --git a/.idea/misc.xml b/.idea/misc.xml
diff --git a/.idea/opendota2.iml b/.idea/opendota2.iml
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/opendota2/__init__.py b/opendota2/__init__.py
@@ -577,7 +577,7 @@ def __get_top_team_matches(low_lim, up_lim):
                 team_matches_ids = [match['match_id'] for match in team_matches]
                 break
             except Exception as e:
-                time.sleep(10)
+                time.sleep(3)
         matches_lst += team_matches_ids
         team_count += 1
         __progressBar(team_count, up_lim - low_lim)
@@ -633,8 +633,7 @@ def get_top_team_matches_dataframe(low_lim, up_lim):
     matches_id_lst, top_teams_rating_dict = __get_top_team_matches(low_lim, up_lim)
     with open("./matches_lst.csv", "w") as matches_lst_file:
         wr = csv.writer(matches_lst_file)
-        for match_id in matches_id_lst:
-            wr.writerow(match_id)
+        wr.writerow(matches_id_lst)
 
     columns = ['Radiant_team_id','Radiant_team_rating', 'Dire_team_id', 'Dire_team_rating',
            'Radiant_hero1', 'Radiant_hero2', 'Radiant_hero3', 'Radiant_hero4', 'Radiant_hero5',
@@ -680,7 +679,7 @@ def get_top_team_matches_dataframe(low_lim, up_lim):
                     df_match_data.loc[count, 'start_time'] = start_time
                     break
             except Exception as e:
-                time.sleep(10)
+                time.sleep(3)
         count += 1
         __progressBar(count, total_len)
     return df_match_data

diff --git a/test/matches_lst.csv b/test/matches_lst.csv
diff --git a/test/test.py b/test/test.py
@@ -3,5 +3,5 @@
 '''Test the api'''
 import opendota2 as op
 
-# get top 100 team matches
-matches_df = op.get_top_team_matches_dataframe(100)
+# get top 10 team matches
+matches_df = op.get_top_team_matches_dataframe(0, 2)