1 line
4.0 KiB
Python
1 line
4.0 KiB
Python
IyBDb3B5cmlnaHQgKEMpIDIwMjQgQ2V0bWl4IE/DnAojIExpY2Vuc2UgQUdQTC0zLjAgb3IgbGF0ZXIgKGh0dHA6Ly93d3cuZ251Lm9yZy9saWNlbnNlcy9hZ3BsKS4KCmZyb20gb2RvbyBpbXBvcnQgXywgZmllbGRzLCBtb2RlbHMKZnJvbSBvZG9vLmV4Y2VwdGlvbnMgaW1wb3J0IEFjY2Vzc0Vycm9yLCBWYWxpZGF0aW9uRXJyb3IKCgpjbGFzcyBDeFRvd2VySmV0U3RhdGUobW9kZWxzLk1vZGVsKToKICAgICIiIkpldCBTdGF0ZXMgcmVwcmVzZW50IHRoZSBkaWZmZXJlbnQgc3RhdGVzIGEgamV0IGNhbiBiZSBpbiBkdXJpbmcgaXRzIGxpZmVjeWNsZSIiIgoKICAgIF9uYW1lID0gImN4LnRvd2VyLmpldC5zdGF0ZSIKICAgIF9kZXNjcmlwdGlvbiA9ICJDZXRtaXggVG93ZXIgSmV0IFN0YXRlIgogICAgX2luaGVyaXQgPSBbImN4LnRvd2VyLnJlZmVyZW5jZS5taXhpbiIsICJjeC50b3dlci5hY2Nlc3MubWl4aW4iXQogICAgX29yZGVyID0gInNlcXVlbmNlLCBpZCIKCiAgICBzZXF1ZW5jZSA9IGZpZWxkcy5JbnRlZ2VyKGRlZmF1bHQ9MTAsIHJlcXVpcmVkPVRydWUpCiAgICBhY3RpdmUgPSBmaWVsZHMuQm9vbGVhbihkZWZhdWx0PVRydWUpCiAgICBjb2xvciA9IGZpZWxkcy5JbnRlZ2VyKCkKICAgIG5vdGUgPSBmaWVsZHMuVGV4dCgpCgogICAgIyBTZXQgZGVmYXVsdCBhY2Nlc3MgbGV2ZWwgdG8gVXNlcgogICAgYWNjZXNzX2xldmVsID0gZmllbGRzLlNlbGVjdGlvbihkZWZhdWx0PSIxIikKCiAgICBkZWYgdW5saW5rKHNlbGYpOgogICAgICAgICIiIgogICAgICAgIERvIG5vdCBhbGxvdyB0byB1bmxpbmsgYSBzdGF0ZQogICAgICAgIGlmIGl0IGlzIHVzZWQgaW4gYW55IGFjdGlvbgogICAgICAgICIiIgogICAgICAgIGFjdGlvbnMgPSBzZWxmLmVudlsiY3gudG93ZXIuamV0LmFjdGlvbiJdLnNlYXJjaCgKICAgICAgICAgICAgWwogICAgICAgICAgICAgICAgInwiLAogICAgICAgICAgICAgICAgInwiLAogICAgICAgICAgICAgICAgKCJzdGF0ZV9mcm9tX2lkIiwgImluIiwgc2VsZi5pZHMpLAogICAgICAgICAgICAgICAgKCJzdGF0ZV90b19pZCIsICJpbiIsIHNlbGYuaWRzKSwKICAgICAgICAgICAgICAgICgic3RhdGVfdHJhbnNpdF9pZCIsICJpbiIsIHNlbGYuaWRzKSwKICAgICAgICAgICAgXQogICAgICAgICkKICAgICAgICBpZiBhY3Rpb25zOgogICAgICAgICAgICByYWlzZSBWYWxpZGF0aW9uRXJyb3IoCiAgICAgICAgICAgICAgICBfKAogICAgICAgICAgICAgICAgICAgICJTb21lIHN0YXRlcyBhcmUgc3RpbGwgdXNlZCBpbiB0aGUgZm9sbG93aW5nIGFjdGlvbnM6ICUoYWN0aW9ucylzIgogICAgICAgICAgICAgICAgICAgICJcbkpldCB0ZW1wbGF0ZXM6ICUodGVtcGxhdGVzKXMiLAogICAgICAgICAgICAgICAgICAgIGFjdGlvbnM9IiwgIi5qb2luKHNldChhY3Rpb25zLm1hcHBlZCgibmFtZSIpKSksCiAgICAgICAgICAgICAgICAgICAgdGVtcGxhdGVzPSIsICIuam9pbihzZXQoYWN0aW9ucy5tYXBwZWQoImpldF90ZW1wbGF0ZV9pZC5uYW1lIikpKSwKICAgICAgICAgICAgICAgICkKICAgICAgICAgICAgKQogICAgICAgIHJldHVybiBzdXBlcigpLnVubGluaygpCgogICAgZGVmIHNldF9zdGF0ZShzZWxmLCBqZXQ9Tm9uZSk6CiAgICAgICAgIiIiU2V0cyB0aGUgc3RhdGUgb2YgdGhlIGpldAoKICAgICAgICBBcmdzOgogICAgICAgICAgICBqZXQgKGN4LnRvd2VyLmpldCk6IEpldCB0byBzZXQgdGhlIHN0YXRlLgogICAgICAgICIiIgogICAgICAgIHNlbGYuZW5zdXJlX29uZSgpCgogICAgICAgICMgVHJ5IHRvIG9idGFpbiBqZXQgZnJvbSBjb250ZXh0IGlmIG5vdCBwcm92aWRlZCBhcyBhbiBhcmd1bWVudAogICAgICAgIGlmIGpldCBpcyBOb25lOgogICAgICAgICAgICBqZXRfaWQgPSBzZWxmLmVudi5jb250ZXh0LmdldCgiamV0X2lkIikKCiAgICAgICAgICAgICMgSnVzdCByZXR1cm4sIG5vIGV4Y2VwdGlvbnMgZm9yIG5vdwogICAgICAgICAgICBpZiBub3QgamV0X2lkOgogICAgICAgICAgICAgICAgcmV0dXJuCgogICAgICAgICAgICBqZXQgPSBzZWxmLmVudlsiY3gudG93ZXIuamV0Il0uYnJvd3NlKGpldF9pZCkKCiAgICAgICAgIyBFbnN1cmUgdGhhdCB0aGUgc3RhdGUgaXMgc2V0IGZvciBhIHNpbmdsZSBqZXQKICAgICAgICBpZiBub3QgamV0IG9yIGxlbihqZXQpID4gMToKICAgICAgICAgICAgcmFpc2UgVmFsaWRhdGlvbkVycm9yKF8oIlN0YXRlIGNhbiBiZSBzZXQgb25seSBmb3IgYSBzaW5nbGUgamV0IikpCgogICAgICAgICMgQ2hlY2sgYWNjZXNzIHRvIHRoZSBqZXQKICAgICAgICBqZXQuY2hlY2tfYWNjZXNzX3JpZ2h0cygicmVhZCIpCiAgICAgICAgamV0LmNoZWNrX2FjY2Vzc19ydWxlKCJ3cml0ZSIpCgogICAgICAgICMgR2V0IHVzZXIgYWNjZXNzIGxldmVsCiAgICAgICAgdXNlcl9hY2Nlc3NfbGV2ZWwgPSBzZWxmLmVudi51c2VyLl9jZXRtaXhfdG93ZXJfYWNjZXNzX2xldmVsKCkKCiAgICAgICAgIyBJZiB1c2VyIGlzIG1hbmFnZXIgYnV0IGlzIG5vdCBhZGRlZCBhcyBhIG1hbmFnZXIgdG8gdGhlIGpldCwKICAgICAgICAjIGhpcyBhY2Nlc3MgbGV2ZWwgaXMgY29uc2lkZXJlZCBhcyB1c2VyLgogICAgICAgICMgTkI6IHJlY29yZCBhY2Nlc3MgaXMgYWxyZWFkeSBjaGVja2VkIGFib3ZlLgogICAgICAgIGlmIHVzZXJfYWNjZXNzX2xldmVsID09ICIyIiBhbmQgc2VsZi5lbnYudXNlciBub3QgaW4gamV0Lm1hbmFnZXJfaWRzOgogICAgICAgICAgICB1c2VyX2FjY2Vzc19sZXZlbCA9ICIxIgoKICAgICAgICAjIENoZWNrIGlmIHVzZXIgYWNjZXNzIGxldmVsIGlzIGVxdWFsIG9yIGdyZWF0ZXIKICAgICAgICBpZiBzZWxmLmFjY2Vzc19sZXZlbCA+IHVzZXJfYWNjZXNzX2xldmVsOgogICAgICAgICAgICByYWlzZSBBY2Nlc3NFcnJvcigKICAgICAgICAgICAgICAgIF8oIllvdSBhcmUgbm90IGFsbG93ZWQgdG8gc2V0IHRoZSAnJShzdGF0ZSlzJyBzdGF0ZSEiLCBzdGF0ZT1zZWxmLm5hbWUpCiAgICAgICAgICAgICkKCiAgICAgICAgIyBCcmluZyB0aGUgamV0IHRvIHRoZSBzdGF0ZQogICAgICAgIGpldC5fYnJpbmdfdG9fc3RhdGUoc2VsZikK |