diff --git a/scrapy_gui/__init__.py b/scrapy_gui/__init__.py index 6802830..426f728 100644 --- a/scrapy_gui/__init__.py +++ b/scrapy_gui/__init__.py @@ -1,2 +1,2 @@ -from .load_selector import load_selector +from .loaders import load_selector, load_response from .browser import open_browser diff --git a/scrapy_gui/load_selector.py b/scrapy_gui/loaders.py similarity index 61% rename from scrapy_gui/load_selector.py rename to scrapy_gui/loaders.py index 8dd0248..fe2bbee 100644 --- a/scrapy_gui/load_selector.py +++ b/scrapy_gui/loaders.py @@ -2,13 +2,15 @@ from .utils_ui.text_viewer import TextViewer from .utils_ui.tools_tab_ui import Queries +from .utils_ui.headers_tab_ui import HeadersViewer import sys class MiniUI(QMainWindow): - def __init__(self, *args, **kwargs): + def __init__(self, *args, response=False, **kwargs): super().__init__(*args, **kwargs) + self.response = response self.init_ui() def init_ui(self): @@ -16,15 +18,19 @@ def init_ui(self): tabs = QTabWidget() self.queries = Queries(main=self) self.source_viewer = TextViewer() + if self.response: + self.headers = HeadersViewer() self.notes = QPlainTextEdit() tabs.addTab(self.queries, 'Tools') tabs.addTab(self.source_viewer, 'Source') tabs.addTab(self.notes, 'Notes') self.setCentralWidget(tabs) - def add_selector(self, selector): + def add_selector(self, selector, response=False): self.queries.update_source(selector) self.source_viewer.setPrettyHtml(selector.text) + if response: + self.headers.update_source(selector) def load_selector(selector): @@ -34,3 +40,12 @@ def load_selector(selector): main.add_selector(selector) main.show() app.exec_() + + +def load_response(response): + print('Shell UI window opened - Close window to regain use of shell') + app = QApplication(sys.argv) + main = MiniUI(response=True) + main.add_selector(response, response=True) + main.show() + app.exec_() diff --git a/scrapy_gui/utils_ui/headers_tab_ui.py b/scrapy_gui/utils_ui/headers_tab_ui.py new file mode 100644 index 0000000..c2c3620 --- /dev/null +++ b/scrapy_gui/utils_ui/headers_tab_ui.py @@ -0,0 +1,6 @@ +from PyQt5.QtWidgets import * +from PyQt5.QtCore import * + + +class HeadersViewer(QWidget): + pass