Abtract the bot code out to Bot
This commit is contained in:
parent
f438f851ea
commit
5b31e2b134
@ -7,6 +7,7 @@ sugar_PYTHON = \
|
|||||||
env.py \
|
env.py \
|
||||||
logger.py \
|
logger.py \
|
||||||
setup.py \
|
setup.py \
|
||||||
|
simulator.py \
|
||||||
TracebackUtils.py \
|
TracebackUtils.py \
|
||||||
util.py
|
util.py
|
||||||
|
|
||||||
|
71
sugar/simulator.py
Normal file
71
sugar/simulator.py
Normal file
@ -0,0 +1,71 @@
|
|||||||
|
import os
|
||||||
|
|
||||||
|
import gtk
|
||||||
|
import gobject
|
||||||
|
|
||||||
|
from sugar.session.TestSession import TestSession
|
||||||
|
from sugar.presence import PresenceService
|
||||||
|
from sugar.p2p import Stream
|
||||||
|
from sugar import util
|
||||||
|
|
||||||
|
_PRESENCE_SERVICE_TYPE = "_presence_olpc._tcp"
|
||||||
|
|
||||||
|
class _SimulatedActivity:
|
||||||
|
def __init__(self):
|
||||||
|
self._id = util.unique_id()
|
||||||
|
|
||||||
|
def get_id(self):
|
||||||
|
return self._id
|
||||||
|
|
||||||
|
class _ShellOwner(object):
|
||||||
|
def __init__(self, nick, color):
|
||||||
|
self._pservice = PresenceService.get_instance()
|
||||||
|
self._color = color
|
||||||
|
self._nick = nick
|
||||||
|
|
||||||
|
def announce(self):
|
||||||
|
props = { 'color': self._color.to_string() }
|
||||||
|
self._service = self._pservice.register_service(self._nick,
|
||||||
|
_PRESENCE_SERVICE_TYPE, properties=props)
|
||||||
|
self._stream = Stream.Stream.new_from_service(self._service)
|
||||||
|
self._stream.register_reader_handler(self._handle_buddy_icon_request, "get_buddy_icon")
|
||||||
|
self._stream.register_reader_handler(self._handle_invite, "invite")
|
||||||
|
|
||||||
|
def _handle_buddy_icon_request(self):
|
||||||
|
return ''
|
||||||
|
|
||||||
|
def _handle_invite(self, issuer, bundle_id, activity_id):
|
||||||
|
return ''
|
||||||
|
|
||||||
|
class Bot:
|
||||||
|
def __init__(self, nick, color):
|
||||||
|
self._nick = nick
|
||||||
|
self._color = color
|
||||||
|
|
||||||
|
os.environ['SUGAR_NICK_NAME'] = self._nick
|
||||||
|
|
||||||
|
def start(self):
|
||||||
|
session = TestSession()
|
||||||
|
session.start()
|
||||||
|
|
||||||
|
PresenceService.start()
|
||||||
|
|
||||||
|
owner = _ShellOwner(self._nick, self._color)
|
||||||
|
owner.announce()
|
||||||
|
|
||||||
|
gobject.timeout_add(1000, self._real_start)
|
||||||
|
|
||||||
|
gtk.main()
|
||||||
|
|
||||||
|
def _real_start(self):
|
||||||
|
pservice = PresenceService.get_instance()
|
||||||
|
|
||||||
|
if not pservice.get_owner().get_color():
|
||||||
|
return True
|
||||||
|
|
||||||
|
activity = _SimulatedActivity()
|
||||||
|
properties = { 'title' : 'OLPC' }
|
||||||
|
activity_type = '_GroupChatActivity_Sugar_redhat_com._udp'
|
||||||
|
service = pservice.share_activity(activity, activity_type, properties)
|
||||||
|
|
||||||
|
return False
|
@ -1,68 +1,7 @@
|
|||||||
#!/usr/bin/python
|
#!/usr/bin/python
|
||||||
import os
|
|
||||||
|
|
||||||
import gtk
|
from sugar.simulator import Bot
|
||||||
import gobject
|
|
||||||
|
|
||||||
from sugar.session.TestSession import TestSession
|
|
||||||
from sugar.presence import PresenceService
|
|
||||||
from sugar.canvas.IconColor import IconColor
|
from sugar.canvas.IconColor import IconColor
|
||||||
from sugar.p2p import Stream
|
|
||||||
from sugar import util
|
|
||||||
|
|
||||||
PRESENCE_SERVICE_TYPE = "_presence_olpc._tcp"
|
bot = Bot('kiu', IconColor('#5E4505,#0F8A0F'))
|
||||||
|
bot.start()
|
||||||
class SimulatedActivity:
|
|
||||||
def __init__(self):
|
|
||||||
self._id = util.unique_id()
|
|
||||||
|
|
||||||
def get_id(self):
|
|
||||||
return self._id
|
|
||||||
|
|
||||||
class ShellOwner(object):
|
|
||||||
def __init__(self):
|
|
||||||
self._pservice = PresenceService.get_instance()
|
|
||||||
self._color = IconColor()
|
|
||||||
self._nick = 'kiu'
|
|
||||||
|
|
||||||
def announce(self):
|
|
||||||
props = { 'color': self._color.to_string() }
|
|
||||||
self._service = self._pservice.register_service(self._nick,
|
|
||||||
PRESENCE_SERVICE_TYPE, properties=props)
|
|
||||||
self._stream = Stream.Stream.new_from_service(self._service)
|
|
||||||
self._stream.register_reader_handler(self._handle_buddy_icon_request, "get_buddy_icon")
|
|
||||||
self._stream.register_reader_handler(self._handle_invite, "invite")
|
|
||||||
|
|
||||||
def _handle_buddy_icon_request(self):
|
|
||||||
return ''
|
|
||||||
|
|
||||||
def _handle_invite(self, issuer, bundle_id, activity_id):
|
|
||||||
return ''
|
|
||||||
|
|
||||||
def start():
|
|
||||||
pservice = PresenceService.get_instance()
|
|
||||||
|
|
||||||
if not pservice.get_owner().get_color():
|
|
||||||
print 'Color not found'
|
|
||||||
return True
|
|
||||||
|
|
||||||
activity = SimulatedActivity()
|
|
||||||
properties = { 'title' : 'OLPC' }
|
|
||||||
activity_type = '_GroupChatActivity_Sugar_redhat_com._udp'
|
|
||||||
service = pservice.share_activity(activity, activity_type, properties)
|
|
||||||
|
|
||||||
return False
|
|
||||||
|
|
||||||
os.environ['SUGAR_NICK_NAME'] = 'kiu'
|
|
||||||
|
|
||||||
session = TestSession()
|
|
||||||
session.start()
|
|
||||||
|
|
||||||
PresenceService.start()
|
|
||||||
|
|
||||||
owner = ShellOwner()
|
|
||||||
owner.announce()
|
|
||||||
|
|
||||||
gobject.timeout_add(1000, start)
|
|
||||||
|
|
||||||
gtk.main()
|
|
||||||
|
Loading…
Reference in New Issue
Block a user