File: simple-agent

package info (click to toggle)
obexd 0.46-1
  • links: PTS, VCS
  • area: main
  • in suites: wheezy
  • size: 2,616 kB
  • sloc: ansic: 25,341; sh: 11,307; python: 423; makefile: 153
file content (131 lines) | stat: -rwxr-xr-x 3,241 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
#!/usr/bin/python

import gobject

import sys
import dbus
import dbus.service
import dbus.mainloop.glib

class Agent(dbus.service.Object):
	def __init__(self, conn=None, obj_path=None):
		dbus.service.Object.__init__(self, conn, obj_path)
		self.pending_auth = False

	@dbus.service.method("org.openobex.Agent",
					in_signature="osssii", out_signature="s")
	def Authorize(self, dpath, device, filename, ftype, length, time):
		global transfers

		self.pending_auth = True
		print "Authorize (%s, %s, %s) Y/n" % (path, device, filename)
		auth = raw_input().strip("\n ")

		if auth == "n" or auth == "N":
			self.pending_auth = False
			raise dbus.DBusException("org.openobex.Error.Rejected: "
									"Not Autorized")

		print "Full filename (including path):"
		self.pending_auth = False

		transfers.append(Transfer(dpath, filename, 0, length))
		return raw_input().strip("\n ")

	@dbus.service.method("org.openobex.Agent",
					in_signature="", out_signature="")
	def Cancel(self):
		print "Authorization Canceled"
		self.pending_auth = False


class Transfer(object):
	def __init__(self, dpath, filename=None, transfered=-1, size=-1):
		self.dpath = dpath
		self.filename = filename
		self.transfered = transfered
		self.size = size

	def update(self, filename=None, transfered=-1, total=-1):
		if filename:
			self.filename = filename
		self.transfered = transfered
		self.size = total

	def cancel(self):
		transfer_iface = dbus.Interface(bus.get_object("org.openobex",
														self.dpath),
										"org.openobex.Transfer")
		transfer_iface.Cancel()

	def __str__(self):
		p = float(self.transfered) / float(self.size) * 100
		return "%s (%s) (%.2f%%)" % (self.filename, self.dpath, p)

	__repr__ = __str__


if __name__ == '__main__':
	global transfers

	def new_transfer(dpath):
		print "new transfer"
		bus.add_signal_receiver(progress,
								dbus_interface="org.openobex.Transfer",
								signal_name="Progress",
								path_keyword="dpath")

	def transfer_completed(dpath, success):
		global transfers

		print "\ntransfer completed => %s" % (success and "Success" or "Fail")
		transfers = [t for t in transfers if t.dpath != dpath]

	def progress(total, current, dpath):
		s = []
		for t in transfers:
			if t.dpath == dpath:
				t.update(None, current, total)
			s.append("%s" % (t))
		sys.stdout.write(" ".join(s) + "\r")
		sys.stdout.flush()


	transfers = []

	dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)

	bus = dbus.SessionBus()
	manager = dbus.Interface(bus.get_object("org.openobex", "/"),
							"org.openobex.Manager")
	bus.add_signal_receiver(new_transfer,
							dbus_interface="org.openobex.Manager",
							signal_name="TransferStarted")

	bus.add_signal_receiver(transfer_completed,
							dbus_interface="org.openobex.Manager",
							signal_name="TransferCompleted")

	path = "/test/agent"
	agent = Agent(bus, path)

	mainloop = gobject.MainLoop()

	manager.RegisterAgent(path)
	print "Agent registered"

	cont = True
	while cont:
		try:
			mainloop.run()
		except KeyboardInterrupt:
			if agent.pending_auth:
				agent.Cancel()
			elif len(transfers) > 0:
				for a in transfers:
					a.cancel()
			else:
				cont = False

	# manager.UnregisterAgent(path)
	# print "Agent unregistered"