Check invalid stations parameter type for Client.select_stations() (#51)

seisman · May 20, 2021 · 0a81031 · 0a81031
1 parent 8c3c96d
commit 0a81031
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/HinetPy/client.py b/HinetPy/client.py
@@ -1151,9 +1151,16 @@ def select_stations(
         0
 
         """
+        stations_selected = []
+
         if stations is None:
-            stations = []
-        stations_selected = stations
+            pass
+        elif isinstance(stations, str):  # stations is a str, i.e., one station
+            stations_selected.append(stations)
+        elif isinstance(stations, list):
+            stations_selected.extend(stations)
+        else:
+            raise ValueError("stations should be either a str or a list.")
 
         # get station list from Hi-net server
         stations_at_server = self.get_station_list(code)