fix bug #2

dev-bearabbit · Feb 19, 2022 · 56d56c3 · 56d56c3
1 parent 4351884
commit 56d56c3
Show file tree

Hide file tree

Showing 8 changed files with 9 additions and 5 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/dist/kbodata-0.1.2.tar.gz b/dist/kbodata-0.1.2.tar.gz
diff --git a/dist/kbodata-0.1.2-py3-none-any.whl → dist/kbodata-0.1.3-py3-none-any.whl b/dist/kbodata-0.1.2-py3-none-any.whl → dist/kbodata-0.1.3-py3-none-any.whl
diff --git a/dist/kbodata-0.1.3.tar.gz b/dist/kbodata-0.1.3.tar.gz
diff --git a/kbodata/__init__.py b/kbodata/__init__.py
@@ -1,4 +1,4 @@
-__version__ = '0.1.0'
+__version__ = '0.1.3'
 
 from kbodata.get.schedule import(
     get_daily_schedule,
@@ -7,7 +7,6 @@
      )
 
 from kbodata.get.game import get_game_data
-
 from kbodata.load.scoreboard import scoreboard_to_DataFrame, scoreboard_to_Dict
 from kbodata.load.batter import batter_to_DataFrame, batter_to_Dict
 from kbodata.load.pitcher import pitcher_to_DataFrame, pitcher_to_Dict
diff --git a/kbodata/parser/config.ini b/kbodata/parser/config.ini
@@ -18,6 +18,7 @@ SK: 7
 넥센: 8
 키움: 8
 서울: 8
+히어로즈: 8
 NC: 9
 KT: 10
 SSG: 11

diff --git a/kbodata/parser/html.py b/kbodata/parser/html.py
@@ -24,8 +24,12 @@ def looking_for_team_names(temp_teams):
     for team in temp_teams:
         temp_team = team.get_text()
         temp_team = temp_team.split(" ")
-        if (temp_team[0] in temp_team_list) == False:
-            temp_team_list.append(temp_team[0])
+        if len(temp_team[0]) == 0:
+            team_name = temp_team[1]
+        else:
+            team_name = temp_team[0]
+        if team_name not in temp_team_list:
+            temp_team_list.append(team_name)
 
     return (temp_team_list[0], temp_team_list[1])
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "kbodata"
-version = "0.1.2"
+version = "0.1.3"
 description = "Scraping Korea Baseball Game information"
 authors = ["Hyeonji-Ryu <[email protected]>"]
 license = "MIT"