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()
