fix for some failing cases

Tommos0 · Sep 27, 2017 · aa6745d · aa6745d
1 parent 9e198b9
commit aa6745d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/pyzenodo/zenodo.py b/pyzenodo/zenodo.py
@@ -36,13 +36,19 @@ def _row_to_version(row):
         }
 
     def get_versions(self):
-        """Get version details from Zenodo webpage (it is not available in the REST api)
-
-        """
+        """Get version details from Zenodo webpage (it is not available in the REST api)"""
         res = requests.get('https://zenodo.org/record/'+self.data['conceptrecid'])
         soup = BeautifulSoup(res.text, 'html.parser')
         version_rows = soup.select('.well.metadata > table.table tr')
-        return [self._row_to_version(row) for row in version_rows]
+        if len(version_rows) == 0:  # when only 1 version
+            return [{
+                'recid': self.data['id'],
+                'name': '1',
+                'doi': self.data['doi'],
+                'date': self.data['created'],
+                'original_version': self.original_version()
+            }]
+        return [self._row_to_version(row) for row in version_rows if len(row.select('td')) > 1]
 
     def original_version(self):
         for identifier in self.data['metadata']['related_identifiers']:

diff --git a/setup.py b/setup.py
@@ -29,6 +29,7 @@
         'License :: OSI Approved :: Apache Software License',
         'Operating System :: OS Independent',
         'Topic :: Software Development :: Libraries :: Python Modules',
+        'Topic :: Scientific/Engineering',
     ],
     packages=find_packages(),
     install_requires=['requests', 'Beautifulsoup4'],