def news_sale_title_find(): #图书促销
global sale_list_link_new
home = "http://www.queshu.com/sale/"
try:
response = urllib2.urlopen(urllib2.Request(home))
soup = BeautifulSoup(response, "html.parser")
for news_sale_detail in soup.find_all(class_="news_sale_detail"):
promotionCompany = news_sale_detail.contents[0].contents[0] #电商名字
promotionDeadline = news_sale_detail.contents[1].string #活动结束时间
for news_sale_title in news_sale_detail.find_all(class_="news_sale_title"): #活动链接
promotionName = news_sale_title.contents[0].contents[0] #活动名称
promotionLink = news_sale_title.contents[0]["href"]
if promotionLink[1:5] == "link":
promotionLink = home[:-5] + promotionLink
if promotionDeadline: #可能为空
promotionDeadline = promotionDeadline.encode('utf-8')
promotionID=promotionLink[-5:]
result = Promotion(
promotionID = promotionID.encode('utf-8'),
promotionCompany = promotionCompany.encode('utf-8'),
promotionName = promotionName.encode('utf-8'),
promotionDeadline = promotionDeadline,
promotionLink = promotionLink.encode('utf-8'))
result.save()
font_13_bold = news_sale_detail.find(href=re.compile("sale_list")) #参加活动图书列表
if font_13_bold:
sale_list_link_new[promotionID] = home[:-6] + font_13_bold["href"]
except urllib2.URLError, e:
print e.reason
def news_sale_title_find(): #活动列表
global sale_list_link_new
home = "http://www.queshu.com/sale/"
try:
response = urllib2.urlopen(urllib2.Request(home))
soup = BeautifulSoup(response, "html.parser")
for news_sale_detail in soup.find_all(class_="news_sale_detail"):
promotionCompany = news_sale_detail.contents[0].contents[0] #电商名字
promotionDeadline = news_sale_detail.contents[1].string #活动结束时间
for news_sale_title in news_sale_detail.find_all(class_="news_sale_title"): #活动链接
promotionName = news_sale_title.contents[0].contents[0] #活动名称
promotionLink = news_sale_title.contents[0]["href"]
if promotionLink[1:5] == "link":
promotionLink = home[:-5] + promotionLink
if promotionDeadline: #活动结束时间可能为空
promotionDeadline = promotionDeadline.encode('utf-8')
promotionID=promotionLink[-5:]
promotionSearchLink = ""
font_13_bold = news_sale_detail.find(href=re.compile("sale_list")) #参加活动图书列表
if font_13_bold:
promotionSearchLink = home[:-6] + font_13_bold["href"]
sale_list_link_new[promotionID] = promotionSearchLink
result = Promotion(
promotionID = promotionID.encode('utf-8'),
promotionCompany = promotionCompany.encode('utf-8'),
promotionName = promotionName.encode('utf-8'),
promotionDeadline = promotionDeadline,
promotionLink = promotionLink.encode('utf-8'),
promotionSearchLink = promotionSearchLink)
result.save()
except urllib2.URLError, e:
print e.reason
def create_promotion():
"""
Create a new promotion
This endpoint will create a promotion based on the data in the request body and save it into the db
"""
check_content_type('application/json')
promotion = Promotion()
promotion.deserialize(request.get_json())
promotion.save()
saved_info = promotion.serialize()
location_url = url_for('get_promotion', promotion_id = promotion.id, _external=True)
return make_response(jsonify(saved_info), status.HTTP_201_CREATED, { 'Location': location_url })
def post(self):
"""
Create a new promotion
This endpoint will create a promotion based on the data in the request body and save it into the db
"""
app.logger.info('Request to Create a Promotion')
check_content_type('application/json')
promotion = Promotion()
app.logger.info('Payload = %s', api.payload)
promotion.deserialize(api.payload)
promotion.save()
app.logger.info('Promotion with new id [%s] saved!', promotion.id)
location_url = api.url_for(PromotionResource,
promotion_id=promotion.id,
_external=True)
return promotion.serialize(), status.HTTP_201_CREATED, {
'Location': location_url
}
class VoucherTests(test.base.BaseAsyncTestCase):
def setUp(self):
super(VoucherTests, self).setUp()
self.admin = test.factories.user()
self.sign_in("admin", "password")
self.shake = Shake(user_id=self.admin.id,
name='promotion-shake',
title='Promotion Shake',
type='group')
self.shake.save()
self.expired_promotion = Promotion(name="Expired Promotion",
membership_months=60,
expires_at=datetime.utcnow() -
timedelta(seconds=50),
promotion_shake_id=0)
self.expired_promotion.save()
self.promotion = Promotion(name="Unit Test Sale",
membership_months=60,
promotion_shake_id=self.shake.id,
expires_at=datetime.utcnow() +
timedelta(seconds=60 * 60 * 24 * 365))
self.promotion.save()
self.used_voucher = Voucher(offered_by_user_id=0,
promotion_id=self.promotion.id,
voucher_key="abc123")
# apply_to_user saves the voucher object (because it touches the
# claimed_by_user_id and dates) and also the user object (by
# updating the is_paid status)
self.used_voucher.apply_to_user(self.admin)
self.unused_voucher = Voucher(offered_by_user_id=0,
claimed_by_user_id=0,
promotion_id=self.promotion.id,
voucher_key="unclaimed")
self.unused_voucher.save()
tornado.httpclient.HTTPClient()
def test_create_account_has_code_field(self):
"""
Create account page should display the discount
code field for new user registration.
"""
self.sign_out()
response = self.fetch_url("/create-account")
self.assertEqual(200, response.code)
self.assertTrue(response.body.find("Discount code:") > -1)
def test_create_account_with_bad_voucher(self):
"""
Submitting an invalid discount code should result in an error.
"""
self.sign_out()
arguments = self._valid_arguments()
arguments["key"] = "ABCDEFGHIJKL"
response = self.post_url("/create-account", arguments=arguments)
self.assertEqual(200, response.code)
self.assertTrue(response.body.find("Invalid discount code") > -1)
def test_create_account_with_unrecognized_voucher(self):
"""
Submitting an invalid discount code should result in an error.
"""
self.sign_out()
arguments = self._valid_arguments()
arguments["key"] = "foobar"
response = self.post_url("/create-account", arguments=arguments)
self.assertEqual(200, response.code)
self.assertTrue(response.body.find("Invalid discount code") > -1)
def test_create_account_with_good_voucher(self):
"""
Submitting a valid discount code should create an account
with appropriate credit applied.
"""
self.sign_out()
arguments = self._valid_arguments()
arguments["key"] = "unclaimed"
# this will create the account, but the redirect doesn't
# carry forward the session info, so sign in separately
# then verify the confirm-account page displays with a greeting
self.post_url("/create-account", arguments=arguments)
self.sign_in(arguments["name"], arguments["password"])
response = self.fetch_url("/confirm-account")
self.assertTrue(
response.body.find("Hello, %s!" % arguments["name"]) > -1)
response = self.fetch_url("/account/settings")
self.assertTrue(response.body.find("5 Years") > -1)
def test_settings_page_with_credit(self):
"""
Validate that our test user account is showing a 5 year
credit on their account settings.
"""
response = self.fetch_url("/account/settings")
self.assertTrue(response.body.find("5 Years") > -1)
def test_redeem_page_with_pro_user(self):
"""
A pro member shouldn't have access to the redeem page.
"""
response = self.fetch_url("/account/redeem")
self.assertTrue(response.body.find("Redeem a Coupon") == -1)
def test_redeem_voucher_with_bad_voucher(self):
self.sign_out()
user = test.factories.user(name="free_user",
email="*****@*****.**",
is_paid=0)
self.sign_in(user.name, "password")
response = self.fetch_url("/account/settings")
# verify this account is currently free
self.assertTrue(
response.body.find("You are currently using a free account.") > -1)
arguments = {"key": "abc123"}
response = self.post_url("/account/redeem", arguments)
self.assertTrue(response.body.find("Invalid") > -1)
def test_redeem_voucher_with_good_voucher(self):
self.sign_out()
user = test.factories.user(name="free_user",
email="*****@*****.**")
user.is_paid = 0
user.save()
self.sign_in(user.name, "password")
response = self.fetch_url("/account/settings")
# verify this account is currently free
self.assertTrue(
response.body.find("You are currently using a free account.") > -1)
arguments = {"key": "unclaimed"}
# this will post and redirect to the settings page which should
# then reflect that we are a paid user with 5 years of credit
response = self.post_url("/account/redeem", arguments)
self.assertTrue(response.body.find("5 Years") > -1)
payments = PaymentLog.where("user_id=%s", user.id)
self.assertEquals(len(payments), 1)
self.assertEquals(payments[0].operation, "redeem")
self.assertEquals(payments[0].status, "credit")
self.assertEquals(payments[0].reference_id, str(self.promotion.id))
self.assertEquals(payments[0].transaction_id, arguments['key'])
self.assertEquals(payments[0].buyer_email, user.email)
self.assertEquals(payments[0].buyer_name, user.name)
# self.assertEquals(payments[0].next_transaction_date, )
voucher = Voucher.get("claimed_by_user_id=%s", user.id)
self.assertEquals(voucher.promotion_id, self.promotion.id)
self.assertEquals(voucher.claimed_by_user_id, user.id)
self.assertEquals(voucher.offered_by_user_id, self.admin.id)
def test_active_promotion_list(self):
promotions = Promotion.active()
self.assertEqual(len(promotions), 1)
self.assertEqual(promotions[0].id, self.promotion.id)
def _valid_arguments(self):
return {
"name": "valid_user",
"password": "******",
"password_again": "asdfasdf",
"email": "*****@*****.**",
"_skip_recaptcha_test_only": True,
}