def testExportSteps(self):
manager = Manager()
stepsStats = dict()
for project in manager.listProjects():
projectFolder = manager.getProjectPath(project.projName)
project = Project(pyworkflow.Config.getDomain(), projectFolder)
project.load()
for prot in project.getRuns():
protName = prot.getClassName()
for step in prot.loadSteps():
stepName = "-".join([
prot.getClassPackageName(), protName,
step.funcName.get()
])
stepSeconds = step.getElapsedTime().total_seconds()
if stepName not in stepsStats:
stepsStats[stepName] = []
stepsStats[stepName].append(stepSeconds)
# average steps
for name in stepsStats:
plugin, protClass, step = name.split("-")
stepStats = stepsStats[name]
mean = sum(stepStats) / len(stepStats)
# DO not send mean values if higher than a threshold to keep chart in a decent visualization range
# We are loosing long duration steps, probably due to large
if mean < 30:
bm = Benchmark(time=mean, name="-".join([protClass, step]))
codespeed.sendData(bm)
def openProject(projectName):
""" Opens a scipion project:
:param projectName: Name of a existing project to open,
or "here" to create a project in the current working dir,
or "last" to open the most recent project
"""
manager = Manager()
projName = os.path.basename(projectName)
# Handle special name 'here' to create a project
# from the current directory
if projName == 'here':
cwd = Config.SCIPION_CWD
if " " in cwd:
print("Projects can't have spaces in the name: %s" % cwd)
sys.exit(1)
print("\nYou are trying to create a project here:", pwutils.cyan(cwd))
if os.listdir(cwd):
print(pwutils.red('\nWARNING: this folder is not empty!!!'))
key = input("\nDo you want to create a project here? [y/N]?")
if key.lower().strip() != 'y':
print("\nAborting...")
sys.exit(0)
else:
print("\nCreating project....")
projName = os.path.basename(cwd)
projDir = os.path.dirname(cwd)
manager.createProject(projName, location=projDir)
elif projName == 'last': # Get last project
projects = manager.listProjects()
if not projects:
sys.exit("No projects yet, cannot open the last one.")
projName = projects[0].projName
projPath = manager.getProjectPath(projName)
projWindow = ProjectWindow(projPath)
projWindow.show()
class ProjectsView(tk.Frame):
def __init__(self, parent, windows, **args):
tk.Frame.__init__(self, parent, bg='white', **args)
self.windows = windows
self.manager = windows.manager
self.root = windows.root
#tkFont.Font(size=12, family='verdana', weight='bold')
bigSize = pwgui.cfgFontSize + 2
smallSize = pwgui.cfgFontSize - 2
fontName = pwgui.cfgFontName
self.projNameFont = tkFont.Font(size=bigSize,
family=fontName,
weight='bold')
self.projDateFont = tkFont.Font(size=smallSize, family=fontName)
self.projDelFont = tkFont.Font(size=smallSize,
family=fontName,
weight='bold')
self.manager = Manager()
# Add the create project button
btnFrame = tk.Frame(self, bg='white')
btn = HotButton(btnFrame,
text=Message.LABEL_CREATE_PROJECT,
font=self.projNameFont,
command=self._onCreateProject)
btn.grid(row=0, column=0, sticky='nw', padx=10, pady=10)
# Add the Import project button
btn = Button(btnFrame,
text=Message.LABEL_IMPORT_PROJECT,
font=self.projNameFont,
command=self._onImportProject)
btn.grid(row=0, column=1, sticky='nw', padx=10, pady=10)
btnFrame.grid(row=0, column=0, sticky='nw')
self.columnconfigure(0, weight=1)
self.rowconfigure(1, weight=1)
text = TaggedText(self, width=40, height=15, bd=0, bg='white')
text.grid(row=1, columnspan=2, column=0, sticky='news')
self.createProjectList(text)
text.setReadOnly(True)
self.text = text
def createProjectList(self, text):
"""Load the list of projects"""
r = 0
text.setReadOnly(False)
text.clear()
parent = tk.Frame(text, bg='white')
parent.columnconfigure(0, weight=1)
colors = ['white', '#EAEBFF']
for i, p in enumerate(self.manager.listProjects()):
try:
project = self.manager.loadProject(p.getName(),
chdir=False,
loadAllConfig=False)
# Add creation time to project info
p.cTime = project.getCreationTime()
# Add if it's a link
p.isLink = project.isLink()
# If it's a link, get the linked folder
if p.isLink:
p.linkedFolder = os.path.realpath(project.path)
frame = self.createProjectLabel(parent, p, color=colors[i % 2])
frame.grid(row=r, column=0, padx=10, pady=5, sticky='new')
r += 1
except Exception, ex:
print "ERROR loading project: %s" % p.getName()
print ex
text.window_create(tk.INSERT, window=parent)
text.bindWidget(parent)
text.setReadOnly(True)
n = len(sys.argv)
if n > 3:
usage("Incorrect number of input parameters")
delete = '--delete' in sys.argv
arg1 = sys.argv[1]
if n > 1 and arg1 != '--delete':
customUserData = arg1
else:
customUserData = pw.Config.SCIPION_USER_DATA
print("Loading projects from:\n %s" % customUserData)
# Create a new project
manager = Manager(workspace=customUserData)
for projInfo in manager.listProjects():
projName = projInfo.getName()
proj = manager.loadProject(projName)
settings = proj.getSettings()
leftTime = proj.getLeftTime()
if (leftTime is not None and leftTime.days < 0):
if delete:
print("Deleting: %s (%s) " % (projName, leftTime))
manager.deleteProject(projName)
else:
print("Should delete: %s (%s)" % (projName, leftTime))
if os.listdir(cwd):
print pwutils.red('\nWARNING: this folder is not empty!!!')
key = raw_input("\nDo you want to create a project here? [y/N]?")
if key.lower().strip() != 'y':
print "\nAborting..."
sys.exit(0)
else:
print "\nCreating project...."
projName = os.path.basename(cwd)
projDir = os.path.dirname(cwd)
proj = manager.createProject(projName, location=projDir)
elif projName == 'last': # Get last project
projects = manager.listProjects()
if not projects:
sys.exit("No projects yet, cannot open the last one.")
projName = projects[0].projName
projPath = manager.getProjectPath(projName)
# try:
projWindow = ProjectWindow(projPath)
# except Exception as e:
# # Print any exception
# print("ERROR: At pw_project.py loading Project %s.\n"
# " Message: %s\n" % (projPath, e))
#
# import traceback
# traceback.print_exc(file=sys.stderr)
#
class ProjectsView(tk.Frame):
_PROJ_CONTAINER = "projectsframe"
def __init__(self, parent, windows, **args):
tk.Frame.__init__(self, parent, bg='white', **args)
self.windows = windows
self.manager = windows.manager
self.root = windows.root
# tkFont.Font(size=12, family='verdana', weight='bold')
bigSize = pwgui.cfgFontSize + 2
smallSize = pwgui.cfgFontSize - 2
fontName = pwgui.cfgFontName
self.projNameFont = tkFont.Font(size=bigSize,
family=fontName,
weight='bold')
self.projDateFont = tkFont.Font(size=smallSize, family=fontName)
self.projDelFont = tkFont.Font(size=smallSize,
family=fontName,
weight='bold')
self.manager = Manager()
self.filter = tk.StringVar()
self.filterBox = None
self.addActionsFrame()
self.columnconfigure(0, weight=1)
self.rowconfigure(1, weight=1)
text = TaggedText(self, width=40, height=15, bd=0, bg='white')
text.grid(row=1, columnspan=2, column=0, sticky='news')
self.createProjectList(text)
text.setReadOnly(True)
self.text = text
self.filterBox.focus_set()
def addActionsFrame(self):
""" Add the "toolbar" for actions like create project, import
project or filter"""
# Add the create project button
bg = "white"
btnFrame = tk.Frame(self, bg=bg)
btn = HotButton(btnFrame,
text=Message.LABEL_CREATE_PROJECT,
font=self.projNameFont,
command=self._onCreateProject)
btn.grid(row=0, column=0, sticky='nw', padx=10, pady=10)
# Add the Import project button
btn = Button(btnFrame,
text=Message.LABEL_IMPORT_PROJECT,
font=self.projNameFont,
command=self._onImportProject)
btn.grid(row=0, column=1, sticky='nw', padx=10, pady=10)
btnFrame.grid(row=0, column=0, sticky='nw')
# Add a filter box
# Add the Import project button
btn = tk.Label(btnFrame, bg=bg, text="Filter:", font=self.projNameFont)
btn.grid(row=0, column=2, sticky='nse', padx=10, pady=10)
self.filterBox = tk.Entry(btnFrame,
font=self.projNameFont,
textvariable=self.filter)
self.filterBox.grid(row=0, column=3, sticky='ne', padx=10, pady=12)
self.filterBox.bind('<Return>', self._onFilter)
self.filterBox.bind('<KP_Enter>', self._onFilter)
def createProjectList(self, text):
"""Load the list of projects"""
r = 0
text.setReadOnly(False)
text.clear()
parent = tk.Frame(text, bg='white', name=self._PROJ_CONTAINER)
parent.columnconfigure(0, weight=1)
colors = ['white', '#EAEBFF']
for i, p in enumerate(self.manager.listProjects()):
try:
# Add creation time to project info
# Add if it's a link
p.index = "index%s" % i
# Consider the filter
if not self._doesProjectMatchFilter(p):
continue
frame = self.createProjectLabel(parent, p, color=colors[i % 2])
frame.grid(row=r, column=0, padx=10, pady=5, sticky='new')
r += 1
except Exception as ex:
print("ERROR loading project: %s" % p.getName())
print(ex)
text.window_create(tk.INSERT, window=parent)
text.bindWidget(parent)
text.setReadOnly(True)
def createProjectLabel(self, parent, projInfo, color):
frame = tk.Frame(parent, bg=color, name=projInfo.index)
# ROW1
# Project name
label = tk.Label(frame,
text=projInfo.projName,
anchor='nw',
bg=color,
justify=tk.LEFT,
font=self.projNameFont,
cursor='hand1',
width=50)
label.grid(row=0, column=0, padx=2, pady=2, sticky='nw')
label.bind('<Button-1>', lambda e: self.openProject(projInfo.projName))
# ROW2
# Timestamp line
dateMsg = '%s%s %s%s' % (
Message.LABEL_MODIFIED, prettyDate(projInfo.mTime),
Message.LABEL_CREATED, prettyTime(projInfo.cTime, time=False))
dateLabel = tk.Label(frame,
text=dateMsg,
font=self.projDateFont,
bg=color)
dateLabel.grid(row=1, column=0, sticky='nw')
# Delete action
delLabel = tk.Label(frame,
text=Message.LABEL_DELETE_PROJECT,
font=self.projDelFont,
bg=color,
cursor='hand1')
delLabel.grid(row=1, column=1, padx=10)
delLabel.bind('<Button-1>', lambda e: self.deleteProject(projInfo))
# Rename action
mvLabel = tk.Label(frame,
text=Message.LABEL_RENAME_PROJECT,
font=self.projDelFont,
bg=color,
cursor='hand1')
mvLabel.grid(row=1, column=2)
mvLabel.bind('<Button-1>',
lambda e: self.renameProject(projInfo.projName))
# ROW3
if projInfo.isLink():
linkMsg = 'link --> ' + projInfo.realPath()
lblLink = tk.Label(frame,
text=linkMsg,
font=self.projDateFont,
bg=color,
fg='grey',
justify=tk.LEFT)
lblLink.grid(row=2, column=0, columnspan=3, sticky='w')
return frame
def createNewProject(self, projName, projLocation):
proj = self.manager.createProject(projName, location=projLocation)
self.createProjectList(self.text)
self.openProject(proj.getShortName())
def _onCreateProject(self, e=None):
projWindow = ProjectCreateWindow("Create project", self)
projWindow.show()
def _onImportProject(self, e=None):
importProjWindow = ProjectImportWindow("Import project", self)
importProjWindow.show()
def _onFilter(self, e=None):
self.createProjectList(self.text)
def _setFocusToList(self, e=None):
self.text.focus_set()
def _doesProjectMatchFilter(self, project):
""" Returns true if the project matches the filter"""
# Lets' use the name anc creation date for now:
searchString = "~".join([
project.getName().lower(),
prettyDate(project.mTime),
prettyTime(project.cTime, time=False)
])
return self.filter.get().lower() in searchString
def importProject(self, projLocation, copyFiles, projName, searchLocation):
self.manager.importProject(projLocation, copyFiles, projName,
searchLocation)
self.createProjectList(self.text)
self.openProject(projName)
def openProject(self, projName):
from subprocess import Popen
script = pw.join(pw.APPS, 'pw_project.py')
Popen([pw.PYTHON, script, projName])
def deleteProject(self, projInfo):
projName = projInfo.projName
if askYesNo(
Message.TITLE_DELETE_PROJECT,
"Project *%s*. " % projName + Message.MESSAGE_DELETE_PROJECT,
self.root):
self.manager.deleteProject(projName)
#Delete the frame
self.text.children[self._PROJ_CONTAINER].children[
projInfo.index].grid_forget()
def renameProject(self, projName):
newName = askString("Rename project %s" % projName, "Enter new name:",
self.root)
if not newName or newName == projName:
return
if self.manager.hasProject(newName):
showError("Rename cancelled",
"Project name already exists: %s" % newName, self.root)
return
self.manager.renameProject(projName, newName)
self.createProjectList(self.text)
