SentryGenericPolicyService-remote 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171
  1. #!/usr/bin/env python
  2. #
  3. # Autogenerated by Thrift Compiler (0.9.3)
  4. #
  5. # DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
  6. #
  7. # options string: py:new_style
  8. #
  9. import sys
  10. import pprint
  11. from urlparse import urlparse
  12. from thrift.transport import TTransport
  13. from thrift.transport import TSocket
  14. from thrift.transport import TSSLSocket
  15. from thrift.transport import THttpClient
  16. from thrift.protocol import TBinaryProtocol
  17. from sentry_generic_policy_service import SentryGenericPolicyService
  18. from sentry_generic_policy_service.ttypes import *
  19. if len(sys.argv) <= 1 or sys.argv[1] == '--help':
  20. print('')
  21. print('Usage: ' + sys.argv[0] + ' [-h host[:port]] [-u url] [-f[ramed]] [-s[sl]] function [arg1 [arg2...]]')
  22. print('')
  23. print('Functions:')
  24. print(' TCreateSentryRoleResponse create_sentry_role(TCreateSentryRoleRequest request)')
  25. print(' TDropSentryRoleResponse drop_sentry_role(TDropSentryRoleRequest request)')
  26. print(' TAlterSentryRoleGrantPrivilegeResponse alter_sentry_role_grant_privilege(TAlterSentryRoleGrantPrivilegeRequest request)')
  27. print(' TAlterSentryRoleRevokePrivilegeResponse alter_sentry_role_revoke_privilege(TAlterSentryRoleRevokePrivilegeRequest request)')
  28. print(' TAlterSentryRoleAddGroupsResponse alter_sentry_role_add_groups(TAlterSentryRoleAddGroupsRequest request)')
  29. print(' TAlterSentryRoleDeleteGroupsResponse alter_sentry_role_delete_groups(TAlterSentryRoleDeleteGroupsRequest request)')
  30. print(' TListSentryRolesResponse list_sentry_roles_by_group(TListSentryRolesRequest request)')
  31. print(' TListSentryPrivilegesResponse list_sentry_privileges_by_role(TListSentryPrivilegesRequest request)')
  32. print(' TListSentryPrivilegesForProviderResponse list_sentry_privileges_for_provider(TListSentryPrivilegesForProviderRequest request)')
  33. print(' TListSentryPrivilegesByAuthResponse list_sentry_privileges_by_authorizable(TListSentryPrivilegesByAuthRequest request)')
  34. print(' TDropPrivilegesResponse drop_sentry_privilege(TDropPrivilegesRequest request)')
  35. print(' TRenamePrivilegesResponse rename_sentry_privilege(TRenamePrivilegesRequest request)')
  36. print('')
  37. sys.exit(0)
  38. pp = pprint.PrettyPrinter(indent = 2)
  39. host = 'localhost'
  40. port = 9090
  41. uri = ''
  42. framed = False
  43. ssl = False
  44. http = False
  45. argi = 1
  46. if sys.argv[argi] == '-h':
  47. parts = sys.argv[argi+1].split(':')
  48. host = parts[0]
  49. if len(parts) > 1:
  50. port = int(parts[1])
  51. argi += 2
  52. if sys.argv[argi] == '-u':
  53. url = urlparse(sys.argv[argi+1])
  54. parts = url[1].split(':')
  55. host = parts[0]
  56. if len(parts) > 1:
  57. port = int(parts[1])
  58. else:
  59. port = 80
  60. uri = url[2]
  61. if url[4]:
  62. uri += '?%s' % url[4]
  63. http = True
  64. argi += 2
  65. if sys.argv[argi] == '-f' or sys.argv[argi] == '-framed':
  66. framed = True
  67. argi += 1
  68. if sys.argv[argi] == '-s' or sys.argv[argi] == '-ssl':
  69. ssl = True
  70. argi += 1
  71. cmd = sys.argv[argi]
  72. args = sys.argv[argi+1:]
  73. if http:
  74. transport = THttpClient.THttpClient(host, port, uri)
  75. else:
  76. socket = TSSLSocket.TSSLSocket(host, port, validate=False) if ssl else TSocket.TSocket(host, port)
  77. if framed:
  78. transport = TTransport.TFramedTransport(socket)
  79. else:
  80. transport = TTransport.TBufferedTransport(socket)
  81. protocol = TBinaryProtocol.TBinaryProtocol(transport)
  82. client = SentryGenericPolicyService.Client(protocol)
  83. transport.open()
  84. if cmd == 'create_sentry_role':
  85. if len(args) != 1:
  86. print('create_sentry_role requires 1 args')
  87. sys.exit(1)
  88. pp.pprint(client.create_sentry_role(eval(args[0]),))
  89. elif cmd == 'drop_sentry_role':
  90. if len(args) != 1:
  91. print('drop_sentry_role requires 1 args')
  92. sys.exit(1)
  93. pp.pprint(client.drop_sentry_role(eval(args[0]),))
  94. elif cmd == 'alter_sentry_role_grant_privilege':
  95. if len(args) != 1:
  96. print('alter_sentry_role_grant_privilege requires 1 args')
  97. sys.exit(1)
  98. pp.pprint(client.alter_sentry_role_grant_privilege(eval(args[0]),))
  99. elif cmd == 'alter_sentry_role_revoke_privilege':
  100. if len(args) != 1:
  101. print('alter_sentry_role_revoke_privilege requires 1 args')
  102. sys.exit(1)
  103. pp.pprint(client.alter_sentry_role_revoke_privilege(eval(args[0]),))
  104. elif cmd == 'alter_sentry_role_add_groups':
  105. if len(args) != 1:
  106. print('alter_sentry_role_add_groups requires 1 args')
  107. sys.exit(1)
  108. pp.pprint(client.alter_sentry_role_add_groups(eval(args[0]),))
  109. elif cmd == 'alter_sentry_role_delete_groups':
  110. if len(args) != 1:
  111. print('alter_sentry_role_delete_groups requires 1 args')
  112. sys.exit(1)
  113. pp.pprint(client.alter_sentry_role_delete_groups(eval(args[0]),))
  114. elif cmd == 'list_sentry_roles_by_group':
  115. if len(args) != 1:
  116. print('list_sentry_roles_by_group requires 1 args')
  117. sys.exit(1)
  118. pp.pprint(client.list_sentry_roles_by_group(eval(args[0]),))
  119. elif cmd == 'list_sentry_privileges_by_role':
  120. if len(args) != 1:
  121. print('list_sentry_privileges_by_role requires 1 args')
  122. sys.exit(1)
  123. pp.pprint(client.list_sentry_privileges_by_role(eval(args[0]),))
  124. elif cmd == 'list_sentry_privileges_for_provider':
  125. if len(args) != 1:
  126. print('list_sentry_privileges_for_provider requires 1 args')
  127. sys.exit(1)
  128. pp.pprint(client.list_sentry_privileges_for_provider(eval(args[0]),))
  129. elif cmd == 'list_sentry_privileges_by_authorizable':
  130. if len(args) != 1:
  131. print('list_sentry_privileges_by_authorizable requires 1 args')
  132. sys.exit(1)
  133. pp.pprint(client.list_sentry_privileges_by_authorizable(eval(args[0]),))
  134. elif cmd == 'drop_sentry_privilege':
  135. if len(args) != 1:
  136. print('drop_sentry_privilege requires 1 args')
  137. sys.exit(1)
  138. pp.pprint(client.drop_sentry_privilege(eval(args[0]),))
  139. elif cmd == 'rename_sentry_privilege':
  140. if len(args) != 1:
  141. print('rename_sentry_privilege requires 1 args')
  142. sys.exit(1)
  143. pp.pprint(client.rename_sentry_privilege(eval(args[0]),))
  144. else:
  145. print('Unrecognized method %s' % cmd)
  146. sys.exit(1)
  147. transport.close()