buy/service/ssgdfs.py

# -*- coding: utf-8 -*-
"""
    demeter service
    name:ssgdfs.py ssgdfs网站抢单业务
    author:rabin
"""
from .__load__ import *

class Ssgdfs(object):
	def init(self, config):
		self.config = config

	def start(self):
		if 'product' in self.config and self.config['product']:
			try:
				# 启动
				self.restart()
				# 开始进入抢购下单核心程序
				self.core()
				# 关闭
				self.quit()
			except BaseException as e:
				print e
				self.start()

	def restart(self):
		# 打开浏览器
		self.open()
		# 登录
		self.login()

	def core(self):
		# 开启多任务
		'''
		task = []
		for v in self.config['product']:
			task.append(gevent.spawn(self.buy, v))
		gevent.joinall(task)
		'''
		for v in self.config['product']:
			self.buy(v)

	def buy(self, config):
		product = Demeter.service('product')
		self.num = 0
		while 1:
			state = self.check(config['link'])
			if state == True:
				try:
					self.num = 0
					print('get' + config['link'])
					product.status(config['id'], 3)
					# 下单
					self.order(config['link'])
					# 支付并获取支付id
					order = product.createOrderId(self.pay(), config['site_id'], config['id'])
					# 支付截图
					pic = self.crop(order, config['id'])
					# 生成订单
					product.order(order, pic, config['site_id'], config['id'])
					product.status(config['id'], 4)
					break
				except BaseException as e:
					print e
					product.status(config['id'], 2)
					continue
			else:
				if self.num == 100:
					print 'restart'
					self.restart()
					break
				else:
					time.sleep(30)
				'''
				self.num = self.num + 1
				#self.close()
				if self.num >= 10:
					self.quit()
					product.status(config['id'], 1)
					print 'kill'
					break
				else:
					time.sleep(30)
				'''

	def check(self, product):
		#r = requests.get(product)
		try:
			self.get(product)
			data = self.driver.page_source
			#print(data)
			# 可以购买
			string = 'btn_direct_buy.gif'
			#data = r.text
			if string in data:
				return True
			else:
				return False
		except BaseException as e:
			self.num = 100
			return False

	def open(self):
		#self.display = Display(visible=0, size=(800, 600))
		#self.display.start()
		dcap = dict(DesiredCapabilities.PHANTOMJS)
		#dcap["phantomjs.page.settings.userAgent"] = (random.choice(USER_AGENTS))
		# 不载入图片
		dcap["phantomjs.page.settings.loadImages"] = False

		# 设置代理
		#service_args = ['--proxy=127.0.0.1:9999','--proxy-type=socks5']
		#self.driver = webdriver.Remote(command_executor=Demeter.config['setting']['phantomjs'], desired_capabilities=DesiredCapabilities.PHANTOMJS)

		self.driver = webdriver.Remote(command_executor=Demeter.config['setting']['phantomjs'], desired_capabilities=DesiredCapabilities.PHANTOMJS)

		#options = webdriver.FirefoxOptions()
		#options.add_argument('user-agent="Mozilla/5.0 (iPod; U; CPU iPhone OS 2_1 like Mac OS X; ja-jp) AppleWebKit/525.18.1 (KHTML, like Gecko) Version/3.1.1 Mobile/5F137 Safari/525.20"')
		#self.driver = webdriver.Chrome(firefox_options=options)

	def login(self):
		self.get(self.config['login_link'])
		#source_code = self.driver.page_source
		self.driver.find_element_by_id('login-id').send_keys(self.config['username'])
		self.driver.find_element_by_id('login-password').send_keys(self.config['password'])
		self.driver.find_element_by_xpath('//input[@alt="login"]').click()
		self.driver.implicitly_wait(5)
		#self.close()

	def order(self, product):
		#self.get('http://cn.ssgdfm.com/shop/common/cookieEnable?ce=null&returnUrl=%2Fshop%2Fproduct%2FproductDetail%3FprdtCode%3D05710004801')
		#self.get(product)
		# 立刻购买
		self.driver.execute_script('directOrderProduct()')
		# 等待加载页面完成
		self.wait('long-sub')
		# 等待弹层关闭
		self.waitNot('pay_popup')
		# 输入下单信息
		self.order_content()
		# 确认订单
		self.driver.execute_script('goPaymentCheck()')
		# 等待加载页面完成
		self.wait('order-agree2')

	def order_content(self):
		# 选择护照信息
		#self.driver.execute_script('setExitInfoHistory("' + self.config['chuguo'] + '")')
		#self.driver.execute_script('setExitInfoHistory("3239636")')
		self.driver.find_element_by_class_name('btn-popup-exitInfo').click()
		exit = self.driver.find_elements_by_class_name('btn-red')
		exitLength = len(exit)-1
		rand = random.randint(0, exitLength)
		exit[rand].click()

		# 使用优惠券
		# 使用积分
		self.driver.find_element_by_xpath('//input[@name="nowUse" and @value="0"]').click()
		self.driver.find_element_by_id('totalUse').click()

	def pay(self):
		# 等待弹层关闭
		self.waitNot('loding_popup')
		# 输入支付信息
		self.pay_content()
		# 确认按钮
		self.driver.execute_script('window.confirm = function(msg) { return true; }');
		# 同意付款
		self.driver.execute_script('$("a.goPayment").click()')
		#self.driver.switch_to_alert().accept()
		#self.wait('qrcode-img-wrapper')
		self.wait('qrcode')
		sleep(2)
		# 获取当前链接，没啥用
		#link = self.driver.execute_script('return location.href')
		# 获取订单号 这里获取不到，直接生成吧
		order = False
		return order

	def pay_content(self):
		# 确认取货处
		self.driver.find_element_by_id('exitDestInfo').click()
		# 同意收集个人信息
		self.driver.find_element_by_id('agree').click()
		# 同意订购
		self.driver.find_element_by_id('agree01').click()
		# 选择支付方式：支付宝21 微信16
		self.driver.find_element_by_xpath('//input[@id="pymntMeansCode" and @value="16"]').click()

	def crop(self, order, id):
		path = Demeter.path + 'runtime/upload/'
		if not File.exists(path):
			File.mkdir(path)
		pic = path + str(id) + '_' + str(order) + '.png'
		self.driver.save_screenshot(pic)
		return pic

	def waitNot(self, name):
		WebDriverWait(self.driver, 100).until_not(lambda x: x.find_element_by_class_name(name).is_displayed())

	def wait(self, name):
		WebDriverWait(self.driver, 100).until(lambda x: x.find_element_by_class_name(name).is_displayed())

	def get(self, link):
		self.driver.get(link)
		self.driver.implicitly_wait(10)

	def close(self):
		self.driver.close()
		#self.driver.close()
		#self.display.stop()

	def quit(self):
		self.driver.quit()
		#self.driver.close()
		#self.display.stop()