コード例 #1
0
ファイル: giftCardSchedule.py プロジェクト: simonqiang/gftest
    def test(self):
        print("running jobs")
        giftCardCodeHelper = GiftCardCodeHelper()
        redisHelper = RedisHelper()
        if not redisHelper.redis_get(giftCardCodeHelper.giftcard_prefix + giftCardCodeHelper.giftcard_flag):

            redisHelper.redis_set_expire(giftCardCodeHelper.giftcard_prefix + giftCardCodeHelper.giftcard_flag, '1',
                                         400)

            for serial_code in Config.SERIAL_CODES:
                try:
                    # check if there is enough serial
                    current_count = redisHelper.redis_scard(
                        giftCardCodeHelper.giftcard_prefix + serial_code + giftCardCodeHelper.giftcard_list)
                    if current_count <= giftCardCodeHelper.giftcard_min_count:
                        # if the index is None or index + generration cout is bigger than 99999999 set index to 1
                        index = redisHelper.redis_get(
                            giftCardCodeHelper.giftcard_prefix + serial_code + giftCardCodeHelper.giftcard_serial_index)
                        if not index:
                            index = 1

                        if (
                                index + giftCardCodeHelper.giftcard_generation_count > giftCardCodeHelper.giftcard_generation_max):
                            index = 1

                        redisHelper.redis_inc(
                            giftCardCodeHelper.giftcard_prefix + serial_code + giftCardCodeHelper.giftcard_serial_index,
                            giftCardCodeHelper.giftcard_generation_count)
                        # insert new serial in redis
                        print('insert new serial in reids')
                        serial_list = giftCardCodeHelper.generate_serial(index,
                                                                         giftCardCodeHelper.giftcard_generation_count,
                                                                         serial_code)
                        redisHelper.redis_sadd(
                            giftCardCodeHelper.giftcard_prefix + serial_code + giftCardCodeHelper.giftcard_list,
                            serial_list)
                except:
                    redisHelper.redis_delete(giftCardCodeHelper.giftcard_prefix + giftCardCodeHelper.giftcard_flag)
                    e = sys.exc_info()[0]
                    traceback.print_exc()
                    return {'error ': str(e)}

            redisHelper.redis_delete(giftCardCodeHelper.giftcard_prefix + giftCardCodeHelper.giftcard_flag)
コード例 #2
0
 def test_generate_pin(self):
     giftCardCodeGenerator = GiftCardCodeHelper()
     for i in range(10000):
         print(giftCardCodeGenerator.generate_pin())
コード例 #3
0
 def test_generate_serial(self):
     giftCardCodeHelper = GiftCardCodeHelper()
     a = 1
     print(giftCardCodeHelper.generate_serial(100, a, 'UPGC1S'))
     a += 100
     print(giftCardCodeHelper.generate_serial(100, a, 'UPGC1S'))