SentryPolicyService-remote 5.7 KB

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