class SignatureTest(VKBaseTest): def setUp(self): self.signature = Signature('secretkey') self.params = {'c': 'a', 'b': 'v', 'a': 'z'} def test_string(self): self.eq(self.signature.string(self.params), 'a=zb=vc=a' + self.signature.key) def test_md5(self): self.eq(self.signature.md5(self.params), md5(self.signature.string(self.params)).hexdigest()) def test_check(self): self.true(self.signature.check(self.params, self.signature.md5(self.params))) self.false(self.signature.check(self.params, 'qwe'))
class VKPayment(BasePayment): def __init__(self, name, goods, secret, callback): self.signature = Signature(secret) self.info = Info(goods) self.order = VKOrder(name, callback) super(VKPayment, self).__init__(goods) def request(self, args): notification_type = args.get('notification_type') try: if not self.signature.check(args, args.pop('sig')): raise SignatureError() item_id, name, count = self.get_name_count(args['item']) if notification_type.startswith(GET_ITEM): return self.info(name, item_id) if notification_type.startswith(ORDER): return self.order(args['order_id'], args['receiver_id'], name, item_id, args['status']) except (ItemFormatError, UnknownItemError, InvalidCountError, CallbackError, SignatureError) as error: return error.response()