diff options
Diffstat (limited to 'module/remote')
| -rw-r--r-- | module/remote/socketbackend/ttypes.py | 13 | ||||
| -rw-r--r-- | module/remote/thriftbackend/pyload.thrift | 11 | ||||
| -rwxr-xr-x | module/remote/thriftbackend/thriftgen/pyload/Pyload-remote | 8 | ||||
| -rw-r--r-- | module/remote/thriftbackend/thriftgen/pyload/Pyload.py | 32 | 
4 files changed, 44 insertions, 20 deletions
| diff --git a/module/remote/socketbackend/ttypes.py b/module/remote/socketbackend/ttypes.py index 6589e5923..682b2b52a 100644 --- a/module/remote/socketbackend/ttypes.py +++ b/module/remote/socketbackend/ttypes.py @@ -207,13 +207,12 @@ class ServerStatus(BaseObject):  		self.reconnect = reconnect  class ServiceCall(BaseObject): -	__slots__ = ['plugin', 'func', 'arguments', 'parseArguments'] +	__slots__ = ['plugin', 'func', 'arguments'] -	def __init__(self, plugin=None, func=None, arguments=None, parseArguments=None): +	def __init__(self, plugin=None, func=None, arguments=None):  		self.plugin = plugin  		self.func = func  		self.arguments = arguments -		self.parseArguments = parseArguments  class ServiceDoesNotExists(Exception):  	__slots__ = ['plugin', 'func'] @@ -238,10 +237,16 @@ class UserData(BaseObject):  		self.permission = permission  		self.templateName = templateName +class UserDoesNotExists(Exception): +	__slots__ = ['user'] + +	def __init__(self, user=None): +		self.user = user +  class Iface:  	def addFiles(self, pid, links):  		pass -	def addPackage(self, name, links, dest): +	def addPackage(self, name, links, dest, password):  		pass  	def call(self, info):  		pass diff --git a/module/remote/thriftbackend/pyload.thrift b/module/remote/thriftbackend/pyload.thrift index 414a1ebf2..a6c0a259c 100644 --- a/module/remote/thriftbackend/pyload.thrift +++ b/module/remote/thriftbackend/pyload.thrift @@ -183,8 +183,7 @@ struct AccountInfo {  struct ServiceCall {      1: PluginName plugin,      2: string func, -    3: optional list<string> arguments, -    4: optional bool parseArguments,  //default False +    3: string arguments, // empty string or json encoded list  }  struct OnlineStatus { @@ -211,6 +210,10 @@ exception FileDoesNotExists{    1: FileID fid  } +exception UserDoesNotExists{ +  1: string user +} +  exception ServiceDoesNotExists{    1: string plugin    2: string func @@ -271,7 +274,7 @@ service Pyload {    // downloads - adding/deleting    list<PackageID> generateAndAddPackages(1: LinkList links, 2: Destination dest), -  PackageID addPackage(1: string name, 2: LinkList links, 3: Destination dest), +  PackageID addPackage(1: string name, 2: LinkList links, 3: Destination dest, 4: string password),    void addFiles(1: PackageID pid, 2: LinkList links),    void uploadContainer(1: string filename, 2: binary data),    void deleteFiles(1: list<FileID> fids), @@ -305,7 +308,7 @@ service Pyload {    //auth    bool login(1: string username, 2: string password), -  UserData getUserData(1: string username, 2:string password), +  UserData getUserData(1: string username, 2:string password) throws (1: UserDoesNotExists ex),    map<string, UserData> getAllUserData(),    //services diff --git a/module/remote/thriftbackend/thriftgen/pyload/Pyload-remote b/module/remote/thriftbackend/thriftgen/pyload/Pyload-remote index f8bcc2863..6ee40092d 100755 --- a/module/remote/thriftbackend/thriftgen/pyload/Pyload-remote +++ b/module/remote/thriftbackend/thriftgen/pyload/Pyload-remote @@ -57,7 +57,7 @@ if len(sys.argv) <= 1 or sys.argv[1] == '--help':    print '   getPackageOrder(Destination destination)'    print '   getFileOrder(PackageID pid)'    print '   generateAndAddPackages(LinkList links, Destination dest)' -  print '  PackageID addPackage(string name, LinkList links, Destination dest)' +  print '  PackageID addPackage(string name, LinkList links, Destination dest, string password)'    print '  void addFiles(PackageID pid, LinkList links)'    print '  void uploadContainer(string filename, string data)'    print '  void deleteFiles( fids)' @@ -350,10 +350,10 @@ elif cmd == 'generateAndAddPackages':    pp.pprint(client.generateAndAddPackages(eval(args[0]),eval(args[1]),))  elif cmd == 'addPackage': -  if len(args) != 3: -    print 'addPackage requires 3 args' +  if len(args) != 4: +    print 'addPackage requires 4 args'      sys.exit(1) -  pp.pprint(client.addPackage(args[0],eval(args[1]),eval(args[2]),)) +  pp.pprint(client.addPackage(args[0],eval(args[1]),eval(args[2]),args[3],))  elif cmd == 'addFiles':    if len(args) != 2: diff --git a/module/remote/thriftbackend/thriftgen/pyload/Pyload.py b/module/remote/thriftbackend/thriftgen/pyload/Pyload.py index 1e2f78b66..3328fb3fc 100644 --- a/module/remote/thriftbackend/thriftgen/pyload/Pyload.py +++ b/module/remote/thriftbackend/thriftgen/pyload/Pyload.py @@ -186,12 +186,13 @@ class Iface(object):      """      pass -  def addPackage(self, name, links, dest): +  def addPackage(self, name, links, dest, password):      """      Parameters:       - name       - links       - dest +     - password      """      pass @@ -1379,22 +1380,24 @@ class Client(Iface):        return result.success      raise TApplicationException(TApplicationException.MISSING_RESULT, "generateAndAddPackages failed: unknown result"); -  def addPackage(self, name, links, dest): +  def addPackage(self, name, links, dest, password):      """      Parameters:       - name       - links       - dest +     - password      """ -    self.send_addPackage(name, links, dest) +    self.send_addPackage(name, links, dest, password)      return self.recv_addPackage() -  def send_addPackage(self, name, links, dest): +  def send_addPackage(self, name, links, dest, password):      self._oprot.writeMessageBegin('addPackage', TMessageType.CALL, self._seqid)      args = addPackage_args()      args.name = name      args.links = links      args.dest = dest +    args.password = password      args.write(self._oprot)      self._oprot.writeMessageEnd()      self._oprot.trans.flush() @@ -2161,6 +2164,8 @@ class Client(Iface):      self._iprot.readMessageEnd()      if result.success is not None:        return result.success +    if result.ex is not None: +      raise result.ex      raise TApplicationException(TApplicationException.MISSING_RESULT, "getUserData failed: unknown result");    def getAllUserData(self, ): @@ -2929,7 +2934,7 @@ class Processor(Iface, TProcessor):      args.read(iprot)      iprot.readMessageEnd()      result = addPackage_result() -    result.success = self._handler.addPackage(args.name, args.links, args.dest) +    result.success = self._handler.addPackage(args.name, args.links, args.dest, args.password)      oprot.writeMessageBegin("addPackage", TMessageType.REPLY, seqid)      result.write(oprot)      oprot.writeMessageEnd() @@ -3218,7 +3223,10 @@ class Processor(Iface, TProcessor):      args.read(iprot)      iprot.readMessageEnd()      result = getUserData_result() -    result.success = self._handler.getUserData(args.username, args.password) +    try: +      result.success = self._handler.getUserData(args.username, args.password) +    except UserDoesNotExists, ex: +      result.ex = ex      oprot.writeMessageBegin("getUserData", TMessageType.REPLY, seqid)      result.write(oprot)      oprot.writeMessageEnd() @@ -4421,12 +4429,14 @@ class addPackage_args(TBase):     - name     - links     - dest +   - password    """    __slots__ = [       'name',      'links',      'dest', +    'password',     ]    thrift_spec = ( @@ -4434,12 +4444,14 @@ class addPackage_args(TBase):      (1, TType.STRING, 'name', None, None, ), # 1      (2, TType.LIST, 'links', (TType.STRING,None), None, ), # 2      (3, TType.I32, 'dest', None, None, ), # 3 +    (4, TType.STRING, 'password', None, None, ), # 4    ) -  def __init__(self, name=None, links=None, dest=None,): +  def __init__(self, name=None, links=None, dest=None, password=None,):      self.name = name      self.links = links      self.dest = dest +    self.password = password  class addPackage_result(TBase): @@ -5254,18 +5266,22 @@ class getUserData_result(TBase):    """    Attributes:     - success +   - ex    """    __slots__ = [       'success', +    'ex',     ]    thrift_spec = (      (0, TType.STRUCT, 'success', (UserData, UserData.thrift_spec), None, ), # 0 +    (1, TType.STRUCT, 'ex', (UserDoesNotExists, UserDoesNotExists.thrift_spec), None, ), # 1    ) -  def __init__(self, success=None,): +  def __init__(self, success=None, ex=None,):      self.success = success +    self.ex = ex  class getAllUserData_args(TBase): | 
