Fix dataframe error when no messages found

Author: jacobrdavis

microSWIFTtelemetry/pull_telemetry.py

@@ -65,7 +65,7 @@ def create_request(
 
 
 def pull_telemetry_as_var(
-    buoy_id: str,
+    buoy_id: str,  # TODO: should this accept a list?
     start_date: datetime,
     end_date: datetime = datetime.utcnow(),
     var_type: str = 'dict',

microSWIFTtelemetry/sbd/compile_sbd.py

@@ -82,17 +82,24 @@ def compile_sbd(
     if var_type == 'pandas':
         df = pandas.DataFrame(data)
         errors = pandas.DataFrame(errors)
+
         if not df.empty:
             to_pandas_datetime_index(df)
         else:
             warnings.warn("Empty DataFrame; if you expected data, make sure "
                           "the `buoy_id` is a valid microSWIFT ID and that "
                           "`start_date` and `end_date` are correct.")
+
+        if not errors.empty:
+            errors = errors.sort_values(by='file_name')
+            errors.reset_index(drop=True, inplace=True)
+
+        #TODO: concatenate dfs?
         return df, errors
 
     if var_type == 'xarray':  # TODO: support for xarray
         raise NotImplementedError('xarray is not supported yet')
-
+    #TODO: should this be 'dataframe' and 'dataset'?
     raise ValueError("var_type can only be 'dict', 'pandas', or 'xarray'")
 
 
@@ -116,7 +123,6 @@ def to_pandas_datetime_index(
     df[datetime_column] = to_datetime(df['datetime'], utc=True)
     df.set_index('datetime', inplace=True)
     df.sort_index(inplace=True)
-    # df.drop(['datetime'], axis=1, inplace=True)
 
 
 def _combine_dict_list(dict_list):

setup.py

@@ -3,7 +3,7 @@
 from setuptools import setup, find_packages
 PACKAGES = find_packages()
 
-# Get version and release info, which is all stored in littlebuoybigwaves/version.py
+# Get version and release info from version.py
 ver_file = os.path.join('microSWIFTtelemetry', 'version.py')
 with open(ver_file) as f:
     exec(f.read())