class ClassUI(QWidget):
def __init__(self):
super().__init__()
self.initUI()
self.re_1 = re.compile(r"^([A-Z]{2,4})(\d{4})")
self.re_2 = re.compile(r"^([A-Z]{2,4})-(\d{4})(\d{2})$")
def initUI(self):
self.layout = QSplitter()
self.in_layout = QVBoxLayout()
self.table = QTableWidget()
self.headers = ['Class', 'Title', 'Amazon', 'Bookstore', 'Edition', 'ISBN', 'Publisher', 'Author', 'Suggested Retail Price', 'Comments', 'Required']
self.table.setColumnCount(len(self.headers))
self.table.setHorizontalHeaderLabels(self.headers)
self.table.cellDoubleClicked.connect(self.tableClicked)
self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
# self.table.verticalHeader().setSectionResizeMode(QHeaderView.Stretch)
self.le = []
for i in range(5):
self.le.append(QLineEdit(self))
self.table.setRowCount(len(self.le))
#self.table.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.submit_btn = QPushButton('Submit', self)
self.submit_btn.clicked.connect(self.submit)
self.add_btn = QPushButton('+', self)
self.add_btn.clicked.connect(self.add_le)
self.rm_btn = QPushButton('-', self)
self.rm_btn.clicked.connect(self.rm_le)
self.btn_layout = QHBoxLayout()
self.in_layout.addWidget(self.submit_btn)
self.btn_layout.addWidget(self.add_btn)
self.btn_layout.addWidget(self.rm_btn)
self.in_layout.addLayout(self.btn_layout)
for l in self.le:
l.textChanged.connect(self.textChanged)
self.in_layout.addWidget(l)
self.regex = re.compile("^[A-Z]{2,4}-\d{4}(?:-\d{2})?$")
lside = QWidget(self)
lside.setLayout(self.in_layout)
self.layout.addWidget(lside)
self.tab = QTabWidget(self)
self.tab.addTab(self.table, 'Books')
self.calendar = WeekView()
self.tab.addTab(self.calendar, 'Calendar')
self.layout.addWidget(self.tab)
l = QVBoxLayout()
l.addWidget(self.layout)
self.setLayout(l)
app_icon = QIcon()
app_icon.addFile('tulsa.jpg')
self.setWindowIcon(app_icon)
self.setGeometry(300, 300, 800, 600)
self.setWindowTitle('Tulsa class info')
self.show()
def add_le(self):
l = QLineEdit(self)
l.textChanged.connect(self.textChanged)
self.in_layout.addWidget(l)
self.le.append(l)
def rm_le(self):
l = self.le.pop()
self.in_layout.removeWidget(l)
l.deleteLater()
l = None
def textChanged(self):
for l in self.le:
t = l.text().upper().replace(' ', '-')
t = re.sub(self.re_1, r"\1-\2", t)
t = re.sub(self.re_2, r"\1-\2-\3", t)
l.setText(t)
if self.regex.match(l.text()):
l.setStyleSheet('color: black')
else:
l.setStyleSheet('color: red')
def tableClicked(self, r, c):
if c == 2 or c == 3:
cell = self.table.item(r, c)
if cell:
url = cell.text()
if url:
QDesktopServices.openUrl(QUrl(url))
def keyPressEvent(self, event):
if event.key() == Qt.Key_Escape:
self.close()
def set_table_item(self, x, y, s):
self.table.setItem(x, y, QTableWidgetItem(s))
def submit(self):
data = [l.text() for l in self.le if self.regex.match(l.text())]
schedule, times = classes.do_stuff(data)
self.calendar.refresh()
self.table.clear()
self.table.setHorizontalHeaderLabels(self.headers)
cur_row = 0
for item in times:
for tup in item['times']:
self.calendar.add_event(tup[0], tup[1], item['name'], item['building'] + ' ' + item['room'])
for key in schedule:
for book in schedule[key]:
self.table.setRowCount(cur_row + 1)
if book['Book Title'] != "No Books Required":
result = search.ddg_crawl(search.ddg_search(book['Book Title'] + ' ' + book['ISBN']))
bkstr = search.bookstore_get_url(book['Book Title'])
self.set_table_item(cur_row, 2, result)
self.set_table_item(cur_row, 3, bkstr)
self.set_table_item(cur_row, 0, key)
self.set_table_item(cur_row, 1, book['Book Title'])
self.set_table_item(cur_row, 4, book['Edition'])
self.set_table_item(cur_row, 5, book['ISBN'])
self.set_table_item(cur_row, 6, book['Publisher'])
self.set_table_item(cur_row, 7, book['Author'])
self.set_table_item(cur_row, 8, book['Publishers Suggested Retail Price'])
self.set_table_item(cur_row, 9, book['Comments'])
self.set_table_item(cur_row, 10, book['Required'])
cur_row += 1
