You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
tde-guidance/mountconfig/SMBShareSelectDialog.py

574 lines
23 KiB

###########################################################################
# SMBShareSelectDialog.py - Dialog for selecting an SMB share on a network#
# ------------------------------ #
# begin : Tue Oct 30 2004 #
# copyright : (C) 2004 by Simon Edwards #
# email : simon@simonzone.com #
# #
###########################################################################
# #
# This program is free software; you can redistribute it and/or modify #
# it under the terms of the GNU General Public License as published by #
# the Free Software Foundation; either version 2 of the License, or #
# (at your option) any later version. #
# #
###########################################################################
from PyTQt.tqt import *
from tdeui import *
from tdecore import *
from tdeio import *
############################################################################
class SMBShareSelectDialog(KDialogBase):
STATUS_IDLE = 0
STATUS_SEARCH_TOP_LEVEL = 1
STATUS_SEARCH = 2
STATUS_RESOLVE = 3
########################################################################
def __init__(self,parent,name=None):
super(SMBShareSelectDialog,self).__init__(parent,name,1,"",KDialogBase.Ok|KDialogBase.Cancel)
self.updatinggui = False
self.resize(600,400)
vbox = self.makeVBoxMainWidget()
hbox = TQHBox(vbox)
hbox.setSpacing(self.spacingHint())
tmplabel = TQLabel(hbox)
tmplabel.setPixmap(UserIcon("hi32-samba"))
hbox.setStretchFactor(tmplabel,0)
self.headinglabel = TQLabel(hbox)
self.headinglabel.setText(i18n("Select a network share"))
hbox.setStretchFactor(self.headinglabel,1)
hbox2 = TQHBox(vbox)
# The main treeview where the action happens.
self.treeview = TDEListView(hbox2)
self.treeview.addColumn("(hidden)")
self.treeview.header().hide()
self.treeview.setRootIsDecorated(True)
self.connect(self.treeview,TQ_SIGNAL("expanded(TQListViewItem *)"),self.slotNodeExpanded)
self.connect(self.treeview,TQ_SIGNAL("selectionChanged(TQListViewItem *)"),self.slotNodeSelected)
self.connect(self.treeview,TQ_SIGNAL("clicked(TQListViewItem *)"),self.slotClicked)
self.dirlister = KDirLister()
self.dirlister.setDirOnlyMode(True)
self.dirlister.setAutoUpdate(False)
self.dirlister.setAutoErrorHandlingEnabled(True,self)
self.connect(self.dirlister,TQ_SIGNAL("newItems(const KFileItemList &)"),self.slotNewItems)
self.connect(self.dirlister,TQ_SIGNAL("completed()"),self.slotDirListCompleted)
self.connect(self.dirlister,TQ_SIGNAL("canceled()"),self.slotDirListCanceled)
self.connect(self.dirlister,TQ_SIGNAL("redirection(const KURL &,const KURL &)"),self.slotDirListRedirection)
self.enableButtonOK(False)
# The "Connect as" part
widget = TQWidget(hbox2)
grid = TQGridLayout(widget,6,4,KDialog.spacingHint())
grid.setRowStretch(5,1)
tmplabel = TQLabel(widget)
tmplabel.setPixmap(UserIcon("hi16-password"))
grid.addWidget(tmplabel,0,0)
self.connectaslabel = TQLabel(widget)
self.connectaslabel.setText("Connect to 'XXX' as:")
grid.addMultiCellWidget(self.connectaslabel,0,0,1,3)
self.guestradio = TQRadioButton(widget)
self.guestradio.setChecked(True)
grid.addWidget(self.guestradio,1,1)
tmplabel = TQLabel(widget)
tmplabel.setText(i18n("Guest"))
grid.addWidget(tmplabel,1,2)
self.connect(self.guestradio,TQ_SIGNAL("stateChanged(int)"),self.slotGuestRadioClicked)
self.userradio = TQRadioButton(widget)
grid.addWidget(self.userradio,2,1)
tmplabel = TQLabel(widget)
tmplabel.setText(i18n("Username:"))
grid.addWidget(tmplabel,2,2)
self.connect(self.userradio,TQ_SIGNAL("stateChanged(int)"),self.slotUserRadioClicked)
self.usernameedit = KLineEdit(widget)
grid.addWidget(self.usernameedit,2,3)
self.connect(self.usernameedit,TQ_SIGNAL("textChanged(const TQString &)"),self.slotUsernameChanged)
tmplabel = TQLabel(widget)
tmplabel.setText(i18n("Password:"))
grid.addWidget(tmplabel,3,2)
self.passwordedit = KLineEdit(widget)
grid.addWidget(self.passwordedit,3,3)
self.reconnectbutton = KPushButton(i18n("Reconnect now"),widget)
grid.addMultiCellWidget(self.reconnectbutton,4,4,1,3)
self.connect(self.reconnectbutton,TQ_SIGNAL("clicked()"),self.slotReconnectClicked)
self.dirlistertimer = None
########################################################################
def choose(self,currenturl):
self.lookupqueue = []
self.selecteditem = None
self.treeview.clear()
self.url_to_list_item_map = {}
# Fill the first level
root_url = KURL("smb:/")
self.rootitem = SMBShareListViewItem(self.treeview, i18n("Network Neighbourhood"), root_url, self)
self.searchurl = currenturl
self._updateConnectGUI()
self.enableButtonOK(False)
self._openDefaultURL()
self.spintimerid = self.startTimer(250)
self.exec_loop()
self.stopResolve()
self.killTimer(self.spintimerid)
if self.result()==self.Accepted:
currenturl = self.selecteditem.getURL()
self.url_to_list_item_map = None
return currenturl
########################################################################
def _openDefaultURL(self):
if self.searchurl is not None:
rc = self.rootitem.selectURL(self.searchurl)
if rc==self.rootitem.OPEN_SUCCESS:
self.currenturl = self.searchurl
self.searchurl = None
self.enableButtonOK(True)
elif rc==self.rootitem.OPEN_FAIL or rc==self.rootitem.OPEN_SUCCESS_INVALID:
self.searchurl = None
########################################################################
def stopResolve(self):
if self.dirlistertimer is not None:
self.killTimer(self.dirlistertimer)
self.dirlister.stop()
for item in self.lookupqueue:
item.cancelResolve()
self.lookupqueue = []
self.searchurl = None # Stop trying to open this URL too.
########################################################################
def setOpen(self,item,open):
if item.isResolved():
TDEListView.setOpen(self.treeview,item,open)
else:
item.startResolve(True)
########################################################################
def appendToResolveQueue(self,item):
if item not in self.lookupqueue:
self.lookupqueue.append(item)
self._startDirLister()
return True
else:
return False
########################################################################
def slotNodeExpanded(self,item):
self.setOpen(item,True)
########################################################################
def slotClicked(self):
if self.treeview.selectedItem() is None:
self.selecteditem = None
self._updateConnectGUI()
self.enableButtonOK(False)
########################################################################
def slotNodeSelected(self,item):
self.selecteditem = item
self._updateConnectGUI()
self.enableButtonOK(item.getLevel()==item.LEVEL_DIR)
if not self.selecteditem.isResolved():
self.selecteditem.startResolve(False)
########################################################################
def slotNewItems(self,items):
for entry in items:
newitem = SMBShareListViewItem(self.lookupqueue[0], str(entry.name()), KURL(entry.url()), self)
self.url_to_list_item_map[str(entry.url().prettyURL())] = newitem
# Notice how I copied the KURL object and TQString (to a python string)
########################################################################
def slotDirListCompleted(self):
item = self.lookupqueue[0]
item.setBusyIcon(False)
del self.lookupqueue[0]
item.resolveComplete()
self._startDirLister()
self._openDefaultURL()
########################################################################
def slotDirListCanceled(self):
self.stopResolve()
########################################################################
def slotDirListRedirection(self,oldUrl,newUrl):
list_item = self.url_to_list_item_map[str(oldUrl.prettyURL())]
list_item.setURL(KURL(newUrl)) # The copy is important.
# Reselect the selected node. (This will force a refresh).
if self.selecteditem is not None:
self.updatinggui = True
self.slotNodeSelected(self.selecteditem)
self.updatinggui = False
########################################################################
def slotUsernameChanged(self,newtext):
self.reconnectbutton.setEnabled(self.usernameedit.text()!="")
########################################################################
def slotReconnectClicked(self):
if self.updatinggui:
return
self.updatinggui = True
if self.selecteditem is None: # Sanity check.
return
# The user wants to change how we connect to this remote machine.
machineitem = self.selecteditem.getMachineItem()
if machineitem is None:
return # Shouldn't happen.
self.stopResolve()
# Grab the URL object before we delete the listviewitem that holds it.
selectedurl = self.selecteditem.getURL()
# Close up the machine item and remove the items under the machine item.
machineitem.unresolve()
# Set the username/password for the machine item.
if self.guestradio.isChecked():
machineitem.getURL().setUser(TQString.null)
machineitem.getURL().setPass(TQString.null)
selectedurl.setUser(TQString.null)
selectedurl.setPass(TQString.null)
else:
machineitem.getURL().setUser(self.usernameedit.text())
machineitem.getURL().setPass(self.passwordedit.text())
selectedurl.setUser(self.usernameedit.text())
selectedurl.setPass(self.passwordedit.text())
self.selecteditem = None
self._updateConnectGUI()
self.searchurl = selectedurl
self._openDefaultURL()
self.updatinggui = False
########################################################################
def _startDirLister(self):
if self.dirlistertimer is None:
# Check the URL lister queue the next the event loop runs.
# Don't get all "recursed up"!
self.dirlistertimer = self.startTimer(0)
########################################################################
def timerEvent(self,event):
KDialogBase.timerEvent(self,event)
if self.spintimerid==event.timerId():
# Spin the current folder icon
if len(self.lookupqueue)!=0:
self.lookupqueue[0].setBusyIcon(True)
elif event.timerId()==self.dirlistertimer:
self.killTimer(self.dirlistertimer)
self.dirlistertimer = None
if self.dirlister.isFinished():
if len(self.lookupqueue)!=0:
self.dirlister.openURL(self.lookupqueue[0].getURL())
########################################################################
def slotGuestRadioClicked(self,state):
if self.updatinggui:
return
self.updatinggui = True
if self.selecteditem is None:
return
if state==TQButton.Off:
self.guestradio.setChecked(True)
self.userradio.setChecked(False)
self.passwordedit.setEnabled(False)
self.usernameedit.setEnabled(False)
selectedurl = self.selecteditem.getURL()
self.reconnectbutton.setEnabled(str(selectedurl.user())!="")
self.updatinggui = False
########################################################################
def slotUserRadioClicked(self,state):
if self.updatinggui:
return
self.updatinggui = True
if state==TQButton.Off:
self.userradio.setChecked(True)
self.guestradio.setChecked(False)
self.passwordedit.setEnabled(True)
self.usernameedit.setEnabled(True)
username = str(self.usernameedit.text())
password = str(self.passwordedit.text())
selectedurl = self.selecteditem.getURL()
if username!="" and password!="" and \
((str(selectedurl.user())!=username) or (str(selectedurl.pass_())!=password)):
self.reconnectbutton.setEnabled(True)
else:
self.reconnectbutton.setEnabled(False)
self.updatinggui = False
########################################################################
def _updateConnectGUI(self):
if self.selecteditem is not None:
selectedurl = self.selecteditem.getURL()
self.guestradio.setEnabled(True)
self.userradio.setEnabled(True)
self.usernameedit.setEnabled(selectedurl.hasUser())
self.passwordedit.setEnabled(selectedurl.hasUser())
self.connectaslabel.setText(i18n("Connect to '%1' as:").arg(selectedurl.host()))
if selectedurl.hasUser():
self.guestradio.setChecked(False)
self.userradio.setChecked(True)
self.usernameedit.setText(selectedurl.user())
self.passwordedit.setText(selectedurl.pass_())
else:
self.guestradio.setChecked(True)
self.userradio.setChecked(False)
self.passwordedit.setText("")
self.usernameedit.setText("")
self.reconnectbutton.setEnabled(False)
else:
self.guestradio.setChecked(True)
self.userradio.setChecked(False)
self.guestradio.setEnabled(False)
self.userradio.setEnabled(False)
self.passwordedit.setEnabled(False)
self.usernameedit.setEnabled(False)
self.connectaslabel.setText(i18n("Connect to 'machine' as:"))
self.guestradio.setChecked(True)
self.userradio.setChecked(False)
self.passwordedit.setText("")
self.usernameedit.setText("")
self.reconnectbutton.setEnabled(False)
############################################################################
class SMBShareListViewItem(TDEListViewItem):
# Return codes for selectURL()
OPEN_SUCCESS = 1
OPEN_SUCCESS_INVALID = 2
OPEN_FAIL = 0
OPEN_BUSY = 3
# Node types.
LEVEL_ROOT = 0
LEVEL_WORKGROUP = 1
LEVEL_MACHINE = 2
LEVEL_DIR = 3 # and deeper.
########################################################################
def __init__(self,parentitem,name,url,smbdialog):
TDEListViewItem.__init__(self,parentitem,name)
if not isinstance(parentitem,SMBShareListViewItem):
self._setIcon(0)
self.setSelectable(False)
else:
self._setIcon(parentitem.depth()+1)
self.setSelectable(parentitem.getLevel()>=self.LEVEL_WORKGROUP)
self.setExpandable(True)
if url.hasPath() and url.path(-1)!="/":
parts = [x for x in str(url.path(-1)).split("/") if x!=""]
self.component = parts[-1].lower()
elif url.hasHost():
self.component = str(url.host()).lower()
else:
self.component = None
self.smbdialog = smbdialog
self.resolved = False
self.url = url
self.autoopen = False
self.animationcounter = 0
########################################################################
def getURL(self):
return self.url
########################################################################
def setURL(self,url):
self.url = url
########################################################################
def getComponent(self):
return self.component
########################################################################
def isResolved(self):
return self.resolved
########################################################################
def startResolve(self,autoopen):
if self.smbdialog.appendToResolveQueue(self):
self.setBusyIcon(True)
self.autoopen = self.autoopen or autoopen
########################################################################
def cancelResolve(self):
self.setBusyIcon(False)
self.autoopen = False
self.resolved = False
while self.childCount()!=0:
self.takeItem(self.firstChild())
self.setOpen(False)
########################################################################
def unresolve(self):
self.cancelResolve()
########################################################################
def getMachineItem(self):
if self.getLevel()<=self.LEVEL_WORKGROUP:
return None
elif self.getLevel()==self.LEVEL_DIR:
return self.parent().getMachineItem()
else:
return self
########################################################################
def _setIcon(self,depth):
if depth==self.LEVEL_ROOT or depth==self.LEVEL_WORKGROUP:
self.setPixmap(0,SmallIcon("network"))
elif depth==self.LEVEL_MACHINE:
self.setPixmap(0,SmallIcon("network_local"))
else:
self.setPixmap(0,SmallIcon("folder"))
########################################################################
def setBusyIcon(self,on):
if on:
self.setPixmap(0,UserIcon("trinity1"))
self.setPixmap(0,UserIcon("trinity"+str(self.animationcounter+1)))
self.animationcounter += 1
self.animationcounter %= 6
else:
self._setIcon(self.depth())
########################################################################
def resolveComplete(self):
self.resolved = True
if self.childCount()==0:
self.setExpandable(False)
else:
if self.autoopen:
self.setOpen(True)
########################################################################
def getLevel(self):
if self.depth()>self.LEVEL_DIR:
return self.LEVEL_DIR
else:
return self.depth()
########################################################################
# This is one of the more nasty pieces of code. It tries to select a given
# URL in the treeview. Opening and resolving the contents of URLs as neccessary
# while at the same time trying not have list everything on the network.
# Another wrinkle is that the treeview contains a level of workgroups while
# a given URL omits the workgroup a jumps directly to the machine name.
def selectURL(self,targeturl):
path = str(targeturl.path(-1))
parts = [x for x in path.split("/") if x!=""]
if targeturl.hasHost():
tmp = [targeturl.host()]
tmp.extend(parts)
parts = tmp
if self.getLevel()==self.LEVEL_ROOT:
# Root item.
# We should first resolve our contents. the Workgroups.
if not self.resolved:
self.startResolve(True)
return self.OPEN_BUSY
else:
if len(parts)==0:
# The URL is really short, and is not selectable.
# So we just say that we couldn't resolve/select it.
return self.OPEN_SUCCESS_INVALID
else:
# OK, the url has some more components. Ask each of the Workgroup items
# to help resolve it.
kid = self.firstChild()
while kid is not None:
rc = kid.selectURL(targeturl)
if rc==self.OPEN_SUCCESS or rc==self.OPEN_SUCCESS_INVALID:
kid.setOpen(True)
return rc
elif rc==self.OPEN_BUSY:
return rc
kid = kid.nextSibling()
return self.OPEN_FAIL
elif self.getLevel()==self.LEVEL_WORKGROUP:
# Workgroup level
if not self.resolved:
self.startResolve(False)
return self.OPEN_BUSY
else:
# Find a child named after the next part of the URL path.
kid = self.firstChild()
partname = parts[0].lower()
while kid is not None:
if kid.getComponent()==partname:
self.setOpen(True)
return kid.selectURL(targeturl)
kid = kid.nextSibling()
return self.OPEN_FAIL
elif self.getLevel()==self.LEVEL_MACHINE:
# Machine level
if len(parts)==1:
# The URL is successfully resolved but is not selectable!
return self.OPEN_SUCCESS_INVALID
else:
# Share level
if len(parts)==self.depth()-1:
self.smbdialog.treeview.setSelected(self,True)
return self.OPEN_SUCCESS
if not self.resolved:
self.startResolve(True)
return self.OPEN_BUSY
else:
# Find a child item that matches the next part of the URL path.
kid = self.firstChild()
partname = parts[self.depth()-1].lower()
while kid is not None:
if kid.getComponent()==partname:
return kid.selectURL(targeturl)
kid = kid.nextSibling()
return self.OPEN_FAIL