Keep old index number of site when parsing .mat file

6 years ago · c391edf755
parent 7e314757ce
commit c391edf755
1 changed files with 2 additions and 0 deletions
--- a/src/data/parse_mat.py
+++ b/src/data/parse_mat.py
@ -151,6 +151,7 @@ def parse_profiles_and_sites(profiles_mat):
    site_counter = 0

    for i, site in enumerate(mat_data["site"]):
+        logger.debug('Processing site {} of {}'.format(i+1, len(mat_data['site'])))

        # Give each site a unique id
        if len(site_rows) == 0 or site_rows[-1]["beach"] != site:
@ -205,6 +206,7 @@ def parse_profiles_and_sites(profiles_mat):
                    profile_rows.append(
                        {
                            "site_id": site_id,
+                            "site_no": i,
                            "lon": lon[0],
                            "lat": lat[0],
                            "profile_type": profile_type,