test/ftp-client

   1 #!/usr/bin/python
   2
   3 from __future__ import absolute_import, print_function, unicode_literals
   4
   5 from optparse import OptionParser
   6 import os.path
   7 import sys
   8 import dbus
   9 import dbus.service
  10 import dbus.mainloop.glib
  11 try:
  12   from gi.repository import GObject
  13 except ImportError:
  14   import gobject as GObject
  15
  16 BUS_NAME='org.bluez.obex'
  17 PATH = '/org/bluez/obex'
  18 CLIENT_INTERFACE='org.bluez.obex.Client1'
  19 SESSION_INTERFACE='org.bluez.obex.Session1'
  20 FILE_TRASNFER_INTERFACE='org.bluez.obex.FileTransfer1'
  21 TRANSFER_INTERFACE='org.bluez.obex.Transfer1'
  22
  23 def parse_options():
  24         parser.add_option("-d", "--device", dest="device",
  25                         help="Device to connect", metavar="DEVICE")
  26         parser.add_option("-c", "--chdir", dest="new_dir",
  27                         help="Change current directory to DIR", metavar="DIR")
  28         parser.add_option("-l", "--list", action="store_true", dest="list_dir",
  29                         help="List the current directory")
  30         parser.add_option("-g", "--get", dest="get_file",
  31                         help="Get FILE", metavar="FILE")
  32         parser.add_option("-p", "--put", dest="put_file",
  33                         help="Put FILE", metavar="FILE")
  34         parser.add_option("-y", "--copy", dest="copy_file",
  35                         help="Copy FILE", metavar="FILE")
  36         parser.add_option("-m", "--move", dest="move_file",
  37                         help="Move FILE", metavar="FILE")
  38         parser.add_option("-n", "--destname", dest="dest_file",
  39                         help="Destination FILE", metavar="FILE")
  40         parser.add_option("-r", "--remove", dest="remove_file",
  41                         help="Remove FILE", metavar="FILE")
  42         parser.add_option("-v", "--verbose", action="store_true",
  43                         dest="verbose")
  44
  45         return parser.parse_args()
  46
  47 class FtpClient:
  48         def __init__(self, session_path, verbose=False):
  49                 self.transferred = 0
  50                 self.transfer_path = None
  51                 self.transfer_size = 0
  52                 self.verbose = verbose
  53                 bus = dbus.SessionBus()
  54                 obj = bus.get_object(BUS_NAME, session_path)
  55                 self.session = dbus.Interface(obj, SESSION_INTERFACE)
  56                 self.ftp = dbus.Interface(obj, FILE_TRASNFER_INTERFACE)
  57                 bus.add_signal_receiver(self.properties_changed,
  58                         dbus_interface="org.freedesktop.DBus.Properties",
  59                         signal_name="PropertiesChanged",
  60                         path_keyword="path")
  61
  62         def create_transfer_reply(self, path, properties):
  63                 self.transfer_path = path
  64                 self.transfer_size = properties["Size"]
  65                 if self.verbose:
  66                         print("Transfer created: %s" % path)
  67
  68         def generic_reply(self):
  69                 if self.verbose:
  70                         print("Operation succeeded")
  71
  72         def error(self, err):
  73                 print(err)
  74                 mainloop.quit()
  75
  76         def properties_changed(self, interface, properties, invalidated, path):
  77                 if path != self.transfer_path:
  78                         return
  79
  80                 if properties['Status'] == 'complete' or \
  81                                 properties['Status'] == 'error':
  82                         if self.verbose:
  83                                 print("Transfer %s" % properties['Status'])
  84                         mainloop.quit()
  85                         return
  86
  87                 if properties["Transferred"] == None:
  88                         return
  89
  90                 speed = (value - self.transferred) / 1000
  91                 print("Transfer progress %d/%d at %d kBps" % (value,
  92                                                         self.transfer_size,
  93                                                         speed))
  94                 self.transferred = value
  95
  96         def change_folder(self, new_dir):
  97                 for node in new_dir.split("/"):
  98                         self.ftp.ChangeFolder(node)
  99
 100         def list_folder(self):
 101                 for i in self.ftp.ListFolder():
 102                         if i["Type"] == "folder":
 103                                 print("%s/" % (i["Name"]))
 104                         else:
 105                                 print("%s" % (i["Name"]))
 106
 107         def put_file(self, filename):
 108                 self.ftp.PutFile(os.path.abspath(filename),
 109                                 os.path.basename(filename),
 110                                 reply_handler=self.create_transfer_reply,
 111                                 error_handler=self.error)
 112
 113         def get_file(self, filename):
 114                 self.ftp.GetFile(os.path.abspath(filename),
 115                                 os.path.basename(filename),
 116                                 reply_handler=self.create_transfer_reply,
 117                                 error_handler=self.error)
 118
 119         def remove_file(self, filename):
 120                 self.ftp.Delete(filename,
 121                                 reply_handler=self.generic_reply,
 122                                 error_handler=self.error)
 123
 124         def move_file(self, filename, destname):
 125                 self.ftp.MoveFile(filename, destname,
 126                                 reply_handler=self.generic_reply,
 127                                 error_handler=self.error)
 128
 129         def copy_file(self, filename, destname):
 130                 self.ftp.CopyFile(filename, destname,
 131                                 reply_handler=self.generic_reply,
 132                                 error_handler=self.error)
 133
 134 if  __name__ == '__main__':
 135
 136         dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
 137
 138         parser = OptionParser()
 139
 140         (options, args) = parse_options()
 141
 142         if not options.device:
 143                 parser.print_help()
 144                 sys.exit(0)
 145
 146         bus = dbus.SessionBus()
 147         mainloop = GObject.MainLoop()
 148
 149         client = dbus.Interface(bus.get_object(BUS_NAME, PATH,),
 150                                                         CLIENT_INTERFACE)
 151
 152         print("Creating Session")
 153         path = client.CreateSession(options.device, { "Target": "ftp" })
 154
 155         ftp_client = FtpClient(path, options.verbose)
 156
 157         if options.new_dir:
 158                 ftp_client.change_folder(options.new_dir)
 159
 160         if options.list_dir:
 161                 ftp_client.list_folder()
 162
 163         if options.get_file:
 164                 ftp_client.get_file(options.get_file)
 165
 166         if options.put_file:
 167                 ftp_client.put_file(options.put_file)
 168
 169         if options.move_file:
 170                 ftp_client.move_file(options.move_file, options.dest_file)
 171
 172         if options.copy_file:
 173                 ftp_client.copy_file(options.copy_file, options.dest_file)
 174
 175         if options.remove_file:
 176                 ftp_client.remove_file(options.remove_file)
 177
 178         mainloop.run()