You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
119 lines
2.6 KiB
119 lines
2.6 KiB
#!/usr/bin/env python
|
|
#
|
|
# Autogenerated by Thrift Compiler (0.11.0)
|
|
#
|
|
# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
|
|
#
|
|
# options string: py
|
|
#
|
|
|
|
import sys
|
|
import pprint
|
|
from urllib.parse import urlparse
|
|
|
|
from PixivSearch.thrift.TestQry import TestQry
|
|
|
|
sys.path.append('/root/PixivSearch')
|
|
|
|
|
|
from thrift.transport import TTransport, TSocket, TSSLSocket, THttpClient
|
|
from thrift.protocol.TBinaryProtocol import TBinaryProtocol
|
|
|
|
|
|
|
|
if len(sys.argv) <= 1 or sys.argv[1] == '--help':
|
|
print('')
|
|
print('Usage: ' + sys.argv[0] + ' [-h host[:port]] [-u url] [-f[ramed]] [-s[sl]] [-novalidate] [-ca_certs certs] [-keyfile keyfile] [-certfile certfile] function [arg1 [arg2...]]')
|
|
print('')
|
|
print('Functions:')
|
|
print(' QryResult qryTest(i32 qryCode)')
|
|
print('')
|
|
sys.exit(0)
|
|
|
|
pp = pprint.PrettyPrinter(indent=2)
|
|
host = 'localhost'
|
|
port = 9090
|
|
uri = ''
|
|
framed = False
|
|
ssl = False
|
|
validate = True
|
|
ca_certs = None
|
|
keyfile = None
|
|
certfile = None
|
|
http = False
|
|
argi = 1
|
|
|
|
if sys.argv[argi] == '-h':
|
|
parts = sys.argv[argi + 1].split(':')
|
|
host = parts[0]
|
|
if len(parts) > 1:
|
|
port = int(parts[1])
|
|
argi += 2
|
|
|
|
if sys.argv[argi] == '-u':
|
|
url = urlparse(sys.argv[argi + 1])
|
|
parts = url[1].split(':')
|
|
host = parts[0]
|
|
if len(parts) > 1:
|
|
port = int(parts[1])
|
|
else:
|
|
port = 80
|
|
uri = url[2]
|
|
if url[4]:
|
|
uri += '?%s' % url[4]
|
|
http = True
|
|
argi += 2
|
|
|
|
if sys.argv[argi] == '-f' or sys.argv[argi] == '-framed':
|
|
framed = True
|
|
argi += 1
|
|
|
|
if sys.argv[argi] == '-s' or sys.argv[argi] == '-ssl':
|
|
ssl = True
|
|
argi += 1
|
|
|
|
if sys.argv[argi] == '-novalidate':
|
|
validate = False
|
|
argi += 1
|
|
|
|
if sys.argv[argi] == '-ca_certs':
|
|
ca_certs = sys.argv[argi+1]
|
|
argi += 2
|
|
|
|
if sys.argv[argi] == '-keyfile':
|
|
keyfile = sys.argv[argi+1]
|
|
argi += 2
|
|
|
|
if sys.argv[argi] == '-certfile':
|
|
certfile = sys.argv[argi+1]
|
|
argi += 2
|
|
|
|
cmd = sys.argv[argi]
|
|
args = sys.argv[argi + 1:]
|
|
|
|
if http:
|
|
transport = THttpClient.THttpClient(host, port, uri)
|
|
else:
|
|
if ssl:
|
|
socket = TSSLSocket.TSSLSocket(host, port, validate=validate, ca_certs=ca_certs, keyfile=keyfile, certfile=certfile)
|
|
else:
|
|
socket = TSocket.TSocket(host, port)
|
|
if framed:
|
|
transport = TTransport.TFramedTransport(socket)
|
|
else:
|
|
transport = TTransport.TBufferedTransport(socket)
|
|
protocol = TBinaryProtocol(transport)
|
|
client = TestQry.Client(protocol)
|
|
transport.open()
|
|
|
|
if cmd == 'qryTest':
|
|
if len(args) != 1:
|
|
print('qryTest requires 1 args')
|
|
sys.exit(1)
|
|
pp.pprint(client.qryTest(eval(args[0]),))
|
|
|
|
else:
|
|
print('Unrecognized method %s' % cmd)
|
|
sys.exit(1)
|
|
|
|
transport.close()
|
|
|