AMi_Image_Analysis/ReadScreen.py at v1.2.2 · LP-CDF/AMi_Image_Analysis · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Wed May 20 09:42:40 2020

Adapted from https://gist.github.com/anonymous/1918b6fec0ab55ae681861e1e36ef754
"""

import os, sys
import csv
from pathlib import Path

from PyQt5.QtWidgets import QTableWidget, QTableWidgetItem

ScreenFile={'MD-PGA':'Md1-51_pga_screen.csv',
            'MD_PACT_Premier':'Md1-36_PACT_Premier_HT.csv',
            'Nextal-MBClassII': 'NeXtal-MbClass-II-Suite.csv',
            'Nextal-Classics-Suite': 'NeXtal-Classics-Suite.csv',
            'Nextal-ClassicsII-Suite': 'NeXtal-ClassicsII-Suite.csv',
            'NeXtal-PEGs-II-Suite':'NeXtal-PEGs-II-Suite.csv',
            'NeXtal-Protein-Complex-Suite':'NeXtal-Protein-Complex-Suite.csv',
            'NeXtal-Nucleix-Suite':'NeXtal-Nucleix-Suite.csv',
            'NeXtal-JCSG-Plus-Suite':'NeXtal-JCSG-Plus-Suite.csv',
            'Jena-JCSG-Plus-Plus':'Jena-JCSGPlusPlus.csv',
            'JBScreen_Classic_HTS_I':'JBScreen_Classic_HTS_I.csv',
            'JBScreen_Classic_HTS_II':'JBScreen_Classic_HTS_II.csv',
            'JBScreen_Classic_1-4':'JBScreen_Classic_1-4.csv',
            'JBScreen_Classic_5-8':'JBScreen_Classic_5-8.csv',
            'Pi-PEG_HTS':'Jena-Pi-PEG_Screen_HTS.csv',
            'HR-PEGRx_HT_screen':'HR-PEGRx_HT_screen.csv',
            'HR-SaltRx_HT_screen':'HR-SaltRx_HT_screen.csv'}

class MyTable(QTableWidget):
    def __init__(self, r, c):
        super().__init__(r, c)

    def open_sheet(self, Screen):
        app_path=os.path.abspath(os.path.dirname(sys.argv[0]))
        path=Path(app_path).joinpath("Screen_Database", ScreenFile[Screen])

        if Path(path).is_file():
            with open(path, newline='') as csv_file:
                my_screen = csv.reader(csv_file, delimiter=',', quotechar='"')
                self.setRowCount(0); self.setColumnCount(10)
                for row_data in my_screen:
                    row = self.rowCount()
                    self.insertRow(row)
                    if len(row_data) > 10:
                        self.setColumnCount(len(row_data))
                    for column, stuff in enumerate(row_data):
                        item = QTableWidgetItem(stuff)
                        self.setItem(row, column, item)
            del my_screen
        else : return False

        del app_path, path