refactoring

This commit is contained in:
frankknoll
2023-06-06 00:43:09 +02:00
parent ee7c5ab3bd
commit c4a9a781eb

View File

@@ -32,12 +32,7 @@ class BatchCodeTableFactory:
return countryBatchCodeTable.loc[country] if country in countryBatchCodeTable.index else self._getEmptyBatchCodeTable(countryBatchCodeTable)
def _getCountryBatchCodeTable(self):
return SummationTableFactory.createSummationTable(
self.dataFrame.groupby(
[
self.dataFrame['COUNTRY'],
self.dataFrame['VAX_LOT']
]))
return SummationTableFactory.createSummationTable(self.dataFrame.groupby(['COUNTRY', 'VAX_LOT']))
def _getEmptyBatchCodeTable(self, countryBatchCodeTable):
return countryBatchCodeTable[0:0].droplevel(0)