"""SME Analytics Ingestion methods.""" def update_label_id(kafka_msg, swb_product): """Populate labelAccount in case of smeAnalyticsIngestion.""" if kafka_msg.message.get('smeAnalyticsIngest'): account_dict = { 'labelAccount': { 'companyCode': kafka_msg.message.get('labelAccountId', None) }, 'subAccount': { 'companyCode': kafka_msg.message.get('subAccountId', None) }, } swb_product['project'].update(account_dict) account_dict['projectId'] = kafka_msg.message.get('projectId', None) account_dict['smeAnalyticsIngest'] = True swb_product.update(account_dict)