code: def set_context(self, cell_type):
          node_names = [node['n'] for node in self.cx['nodes']]
          res_expr = context_client.get_protein_expression(node_names,
                                                           [cell_type])
          res_mut = context_client.get_mutations(node_names,
                                                 [cell_type])
          res_expr = res_expr.get(cell_type)
          res_mut = res_mut.get(cell_type)
          if not res_expr:
              msg = 'Could not get protein expression for %s cell type.' % \
                    cell_type
              logger.warning(msg)
          if not res_mut:
              msg = 'Could not get mutational status for %s cell type.' % \
                    cell_type
              logger.warning(msg)
          if not res_expr and not res_mut:
              return
          self.cx['networkAttributes'].append({'n': 'cellular_context',
                                               'v': cell_type})
          counter = 0
          for node in self.cx['nodes']:
              amount = res_expr.get(node['n'])
              mut = res_mut.get(node['n'])
              if amount is not None:
                  node_attribute = {'po': node['@id'],
                                    'n': 'expression_amount',
                                    'v': int(amount)}
                  self.cx['nodeAttributes'].append(node_attribute)
              if mut is not None:
                  is_mutated = 1 if mut else 0
                  node_attribute = {'po': node['@id'],
                                    'n': 'is_mutated',
                                    'v': is_mutated}
                  self.cx['nodeAttributes'].append(node_attribute)
              if mut is not None or amount is not None:
                  counter += 1
          logger.info('Set context for %d nodes.' % counter)