def query_user_product(self): u = self.session.query(User).filter(User.id==self.username).one() self.assertEqual([p.upper() for p in self.products], [p.id for p in u.products]) self.assertEqual([up for up in u.user_products], [dbman.query_user_product(self.username, p, session=self.session) for p in self.products])
def query_user_product(self): u = self.session.query(User).filter(User.id == self.username).one() self.assertEqual([p.upper() for p in self.products], [p.id for p in u.products]) self.assertEqual([up for up in u.user_products], [ dbman.query_user_product(self.username, p, session=self.session) for p in self.products ])
def execute(self, context): logging.info('Creating squash schedule for user ', self.username) session = context.session try: user_product = query_user_product(self.username, 'squashcity', session) if user_product is None: return user_not_registered() squashcity_req = SquashCityRequest(request_type = 'SQUASHCITY', user_product=user_product, start_date=self.d_from, end_date=self.d_to) session.add(squashcity_req) session.commit() return success() except: print "Exception in SquashCity.execute: ", sys.exc_info() return retry_later()
def execute(self, context): logging.info('Creating squash schedule for user ', self.username) session = context.session try: user_product = query_user_product(self.username, 'squashcity', session) if user_product is None: return user_not_registered() squashcity_req = SquashCityRequest(request_type='SQUASHCITY', user_product=user_product, priority=user_product.priority, start_date=self.d_from, end_date=self.d_to) session.add(squashcity_req) session.commit() return success() except: print "Exception in SquashCity.execute: ", sys.exc_info() return retry_later()