diff --git a/database/protocol.py b/database/protocol.py index 2316a7f3d9924a6013a1c86a581f671106c52e9e..10886a71d98c05c5a483e5f87f5a108785ac0989 100644 --- a/database/protocol.py +++ b/database/protocol.py @@ -74,9 +74,19 @@ class Protocol(object): if not indexes: return None if len(indexes) > 1: - return None + self.resolve_duplicates(year, indexes) + return self._remaped[indexes[0]] + def resolve_duplicates(self, year, indexes): + ''' + Transforms a dbcolumn that gets the data from the same header to a denormalization of the first column. + ''' + original_dbcolumn = self.dbcolumn_from_target(self._remaped[indexes[0]])[0] + + for i in range(1, len(indexes)): + self._dataframe.loc[indexes[i], year] = '~' + original_dbcolumn + def original_from_target(self, name, year): '''Gets original column from target column and a year Input example: **{'name': 'CEBMA015N0', 'year': '2015'}