Source code for ibllib.graphic

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# @Author: Niccolò Bonacchi
# @Date: Sunday, February 3rd 2019, 11:59:56 am
import tkinter as tk
from tkinter import simpledialog
from tkinter import messagebox





[docs]def numinput(title, prompt, default=None, minval=None, maxval=None, nullable=False, askint=False): root = tk.Tk() root.withdraw() ask = simpledialog.askinteger if askint else simpledialog.askfloat ans = ask( title, prompt, initialvalue=default, minvalue=minval, maxvalue=maxval) if ans == 0: return ans elif not ans and not nullable: return numinput( title, prompt, default=default, minval=minval, maxval=maxval, nullable=nullable, askint=askint) return ans
[docs]def strinput(title, prompt, default='COM', nullable=False): """ Example: >>> strinput("RIG CONFIG", "Insert RE com port:", default="COM") """ import tkinter as tk from tkinter import simpledialog root = tk.Tk() root.withdraw() ans = simpledialog.askstring(title, prompt, initialvalue=default) if (ans is None or ans == '' or ans == default) and not nullable: return strinput(title, prompt, default=default, nullable=nullable) else: return ans
[docs]def login(title='Enter Credentials', default_username=None, default_passwd=None, add_fields=None): """ Dialog box prompting for username and password. :param title: Window title :param default_username: default field for username :param default_passwd: default field for password :param add_fields: list of new fields to be added to the dialog :return: """ class Toto: def __init__(self, root, default_username=None, default_passwd=None, title=None, add_fields=None): self.add_fields = add_fields or [] self.var1 = tk.StringVar() self.root = root # self.root.geometry('300x160') self.root.title(title) # frame for window margin self.parent = tk.Frame(self.root, padx=10, pady=10) self.parent.pack(fill=tk.BOTH, expand=True) # entrys with not shown text self.add_entries = [] for fname in self.add_fields: self.add_entries.extend([self.make_entry(self.parent, fname + ":", 42, show="")]) self.user = self.make_entry(self.parent, "User name:", 42, show='', default=default_username) self.password = self.make_entry(self.parent, "Password:", 42, show="*", default=default_passwd) # button to attempt to login self.button = tk.Button(self.parent, borderwidth=4, text="Login", width=42, pady=8, command=self.get_value) self.button.pack(side=tk.BOTTOM) self.user.focus_set() self.USR = None self.MDP = None self.root.bind('<Return>', self.push_enter) # do not reproduce vim behaviour self.root.protocol("WM_DELETE_WINDOW", self.cancel_login) def make_entry(self, _, caption, width=None, default='', **options): tk.Label(self.parent, text=caption).pack(side=tk.TOP) entry = tk.Entry(self.parent, **options) if width: entry.config(width=width) entry.pack(side=tk.TOP, padx=10, fill=tk.BOTH) if default: entry.insert(0, default) return entry def push_enter(self, _): self.get_value() def get_value(self): self.USR = self.user.get() self.MDP = self.password.get() self.ADD = [] for entry in self.add_entries: self.ADD.extend([entry.get()]) self.root.destroy() self.root.quit() def cancel_login(self): self.USR = None self.MDP = None self.ADD = [] for entry in self.add_entries: self.ADD.extend([None]) self.root.destroy() self.root.quit() root = tk.Tk() toto = Toto(root, title=title, default_passwd=default_passwd, default_username=default_username, add_fields=add_fields) root.mainloop() return [toto.USR] + [toto.MDP] + toto.ADD
[docs]def multi_input(title='Enter Credentials', add_fields=None, defaults=None): class Toto: def __init__(self, root, title=None, add_fields=None, defaults=None): self.fields_to_add = add_fields or [] if defaults is None or len(defaults) != len(add_fields): self.defaults = [None for x in self.fields_to_add] else: self.defaults = defaults self.var1 = tk.StringVar() self.root = root # self.root.geometry('300x160') self.root.title(title) # frame for window margin self.parent = tk.Frame(self.root, padx=10, pady=10) self.parent.pack(fill=tk.BOTH, expand=True) # entrys with not shown text self.add_entries = [] for fname, fdef in zip(self.fields_to_add, self.defaults): self.add_entries.extend( [self.make_entry(self.parent, fname + ":", 42, show="", default=fdef)]) # button to attempt to login self.button = tk.Button(self.parent, borderwidth=4, text="Submit", width=42, pady=8, command=self.get_value) self.button.pack(side=tk.BOTTOM) self.root.bind('<Return>', self.push_enter) # do not reproduce vim behaviour self.root.protocol("WM_DELETE_WINDOW", self.cancel_login) def make_entry(self, _, caption, width=None, default='', **options): tk.Label(self.parent, text=caption).pack(side=tk.TOP) entry = tk.Entry(self.parent, **options) if width: entry.config(width=width) entry.pack(side=tk.TOP, padx=10, fill=tk.BOTH) if default: entry.insert(0, default) return entry def push_enter(self, _): self.get_value() def get_value(self): self.ADD = [] for entry in self.add_entries: self.ADD.extend([entry.get()]) self.root.destroy() self.root.quit() def cancel_login(self): self.ADD = [] for entry in self.add_entries: self.ADD.extend([None]) self.root.destroy() self.root.quit() root = tk.Tk() toto = Toto(root, title=title, add_fields=add_fields, defaults=defaults) root.mainloop() return toto.ADD
# from ibllib.misc import login # a, b =login.login(default_passwd='tutu', default_username='turluser') # a, b =login.login(default_passwd='tutu', default_username='turluser', title='supertitre') # a, b, add1, add2 =login.login(default_passwd='tutu', default_username='turluser', # title='supertitre', add_fields=['tuasdf', 'adfasdf']) # title = 'Recording site' # fields = ['X (float):', 'Y (float):', 'Z (flaot):', 'D (float):', # 'Angle (10 or 20):', 'Origin (bregma or lambda):'] # defaults = [None, None, None, None, '10', 'bregma'] # types = [float, float, float, float, int, str] # bla = multi_input(title=title, add_fields=fields, defaults=defaults) # try: # out = [t(x) for x, t in zip(bla, types)] # except Exception: # bla = multi_input(title=title, add_fields=fields, defaults=defaults) # print(bla)