def initActions(self):
collection = self.actionCollection()
self.quitAction = KStdAction.quit(self.hide, collection)
self.newEntityWindowAction = NewEntityWindowAction(self.slotNewEntityWindow,
collection)
self.newRadioWindowAction = NewRadioWindowAction(self.slotNewRadioWindow,
collection)
self.newRtorrentWindowAction = NewRtorrentWindowAction(self.slotNewRtorrentWindow,
collection)
class MainWindow(BaseMainWindow, MainDropCatcher):
def __init__(self, parent):
BaseMainWindow.__init__(self, parent, 'ToolboxMainWindow')
self.label = QLabel('toobox', self)
self.setCentralWidget(self.label)
self.app.main_window = self
self.initActions()
self.initMenus()
self.initToolbar()
self.setAcceptDrops(True)
self.connect(self.app,
PYSIGNAL('UrlHandled'), self.url_handled)
#self.connect(self, SIGNAL('quit()'), self.hide)
self._child_windows = dict()
if os.environ.has_key('DEBUG') and os.environ['DEBUG'] == 'nosystray':
self.systray = self
else:
self.systray = MySytemTray(self)
systray_menu = self.systray.contextMenu()
self.newEntityWindowAction.plug(systray_menu)
self.newRadioWindowAction.plug(systray_menu)
def initActions(self):
collection = self.actionCollection()
self.quitAction = KStdAction.quit(self.hide, collection)
self.newEntityWindowAction = NewEntityWindowAction(self.slotNewEntityWindow,
collection)
self.newRadioWindowAction = NewRadioWindowAction(self.slotNewRadioWindow,
collection)
self.newRtorrentWindowAction = NewRtorrentWindowAction(self.slotNewRtorrentWindow,
collection)
def initMenus(self):
mainmenu = KPopupMenu(self)
self.newEntityWindowAction.plug(mainmenu)
self.newRadioWindowAction.plug(mainmenu)
self.newRtorrentWindowAction.plug(mainmenu)
self.quitAction.plug(mainmenu)
menubar = self.menuBar()
menubar.insertItem('&Main', mainmenu)
def initToolbar(self):
toolbar = self.toolBar()
self.newEntityWindowAction.plug(toolbar)
self.newRadioWindowAction.plug(toolbar)
self.newRtorrentWindowAction.plug(toolbar)
self.quitAction.plug(toolbar)
def _slotNewWindow(self, winclass):
count = 1
name = '%s-%d' % (winclass.__name__, count)
while self._child_windows.has_key(name):
count += 1
name = '%s-%d' % (winclass.__name__, count)
win = winclass(self, name=name)
win.show()
self._child_windows[name] = win
def slotNewEntityWindow(self):
self._slotNewWindow(MainEntityWindow)
def slotNewRadioWindow(self):
self._slotNewWindow(BaseRadioWindow)
def slotNewRtorrentWindow(self):
self._slotNewWindow(BaseRtorrentWindow)
def _make_youtube_dl_filename(self, youtubeid, title):
title = ''.join(x in string.ascii_letters + string.digits and x or '_' for x in title)
title = title.replace(' ', '_')
# determine filename
filename = '%s-%s.flv' % (title, youtubeid)
return filename
def url_handled(self):
urls = self.app.urlhandler.completed_urls()
for url in urls:
if url.host.endswith('youtube.com'):
download = False
main = dict()
data = self.app.urlhandler.retrieve_data(url)
if data.has_key('entityid') and data['entityid'] is None:
entity = self.app.db.Entity()
entity.name = data['title']
entity.type = 'youtube-video'
entity.desc = 'Youtube video'
# I don't seem to be able to add the relations to the object
# without flushing the session to get the new entityid
self.app.db.session.flush()
entityid = entity.entityid
EntityExtraField = self.app.db.EntityExtraField
entity.extfieldlist.append(EntityExtraField(entityid, 'youtubeid',data['youtubeid']))
# default to not there
entity.extfieldlist.append(EntityExtraField(entityid, 'local-copy', False))
# normalize title similar to youtube-dl
filename = self._make_youtube_dl_filename(data['youtubeid'], data['title'])
# add filename to extras
entity.extfieldlist.append(EntityExtraField(entityid, 'local-filename', filename))
# create entity
self.app.db.session.save(entity)
# do we need to flush()?
#self.app.db.session.flush()
download = True
else:
entityid = data['entityid']
db = self.app.db
eef = db.EntityExtraField
query = db.session.query(eef)
eef_filter = query.filter(eef.entityid == entityid)
query = eef_filter.filter(eef.fieldname == 'local-copy')
lc = bool(int(query.one().value))
print "in mainwin, lc is", lc, type(lc)
if not lc:
download = True
query = eef_filter.filter(eef.fieldname == 'local-filename')
filename = query.one().value
if download:
# download youtube video
flv_url = data['flv_url']
print "download True", filename, flv_url
self.app.filehandler.download(data['flv_url'], filename)
msg = "handled url %s" % ',\n '.join(list(urls))
KMessageBox.information(self, msg)
def show(self):
for name, window in self._child_windows.items():
window.show()
BaseMainWindow.show(self)
def hide(self):
if self.systray is self:
BaseMainWindow.close(self)
else:
for name, window in self._child_windows.items():
window.hide()
BaseMainWindow.hide(self)
