Python实现截图生成符合markdown的链接

class ImgFrame(QMainWindow):
    def __init__(self):
        super().__init__()
        self.http_url = None
        self.markdown_url = None
        self.clipboard = None
        self.img = None
        # 初始化数据库
        self.db = init_db()
        self.init_ui()

    def init_ui(self):
        self.setGeometry(300, 300, 500, 500)
        self.setWindowTitle('MarkDown-Img')
        widget = QWidget()
        setupAction = QAction(QIcon('setup.png'), '设置', self)
        setupAction.setStatusTip('Exit application')
        setupAction.triggered.connect(self.a)
        menubar = QMenuBar(self)
        menubar.setGeometry(QtCore.QRect(0, 0, 251, 23))
        menubar.setObjectName("menubar")
        setup = menubar.addMenu('系统')

        setup.addAction(setupAction)
        menubar.setVisible(True)
        menubar.setNativeMenuBar(False)
        self.setMenuBar(menubar)
        self.img = QLabel()
        layout = QVBoxLayout()
        layout.addWidget(markdown_widget(self))
        layout.addWidget(url_widget(self))
        layout.addWidget(self.img)
        layout.setAlignment(Qt.AlignCenter)

        self.clipboard = QApplication.clipboard()
        self.clipboard.dataChanged.connect(self.paste)

        widget.setLayout(layout)
        self.setCentralWidget(widget)
        self.show()

def init_db():
    connect = sqlite3.connect('markdown-img.db')
    global conn
    #全局变量conn
    conn = connect
    cursor = connect.cursor()
    cursor.execute(sql)
    #返回游标
    return cursor

class secondFrame(QWidget):
    def __init__(self, db):
        super().__init__()
        # self.init_ui()
        self.db = db
        self.resize(400, 100)
        self.setWindowTitle('存储设置')

        formlayout = QFormLayout()
        storageLabel = QLabel("存储")
        self.storageBox = QComboBox()
        self.storageBox.addItems(['阿里OSS', '七牛Kodo'])

        self.endpointLabel = QLabel("endpoint")
        self.endpointLineEdit = QLineEdit("")
        self.endpointLineEdit.setStyleSheet("width:200px")

        self.qntLabel = QLabel("七牛域名")
        self.qnLineEdit = QLineEdit("")
        self.qnLineEdit.setStyleSheet("width:200px")

        keyLabel = QLabel("access_key")
        self.keyLineEdit = QLineEdit("")
        self.keyLineEdit.setStyleSheet("width:350px")

        secretLabel = QLabel("secret_key")

        self.secretLineEdit = QLineEdit()
        self.secretLineEdit.setStyleSheet("width:350px")
        self.secretLineEdit.setText('')

        bucketLabel = QLabel("bucket_name")
        self.bucketLineEdit = QLineEdit("")

        confirmButton = QPushButton("确定")

        formlayout.addRow(storageLabel, self.storageBox)
        formlayout.addRow(bucketLabel, self.bucketLineEdit)
        formlayout.addRow(self.endpointLabel, self.endpointLineEdit)
        formlayout.addRow(self.qntLabel, self.qnLineEdit)
        self.qntLabel.setVisible(False)
        self.qnLineEdit.setVisible(False)

        formlayout.addRow(keyLabel, self.keyLineEdit)
        formlayout.addRow(secretLabel, self.secretLineEdit)
        formlayout.addRow(confirmButton)

        self.storageBox.currentIndexChanged[int].connect(self.changeLabel)
        confirmButton.clicked.connect(self.confirm)

        self.setLayout(formlayout)
        self.setVisible(True)

    setupAction = QAction(QIcon('setup.png'), '设置', self)
    setupAction.setStatusTip('Exit application')
    setupAction.triggered.connect(self.openSecondFrame)
    def openSecondFrame(self):
        self.frame = secondFrame(self.db)
        self.frame.show()        

    self.clipboard.dataChanged.connect(self.paste)
    def paste(self):
        data = self.clipboard.mimeData()
        if data.hasImage():
            print(data.formats())
            pixmap = self.clipboard.pixmap()
            height = pixmap.height()
            width = pixmap.width()
            if height > 300 and width > 300:
                self.img.setPixmap(shrink_img(pixmap))
            else:
                self.img.setPixmap(pixmap)
            fileName = 'ink_' + ''.join(str(uuid.uuid1()).split('-')) + '.png'
            self.clipboard.pixmap().save(fileName, 'PNG')
            urls = generate_url(self.upload(fileName))
            print(urls)
            self.img.setScaledContents(True)
            self.markdown_url.setText(urls['markdown_url'])
            self.http_url.setText(urls['http_url'])
            pyperclip.copy(urls['markdown_url'])
    def shrink_img(pixmap):
        scale = 0.3
        height = pixmap.height()
        width = pixmap.width()
        shrink_height = int(height * scale)
        shrink_width = int(width * scale)
        size = QSize(shrink_width, shrink_height)
        image = pixmap.toImage()
        return QPixmap.fromImage(image.scaled(size, Qt.IgnoreAspectRatio))

  def upload(self, filename):
        type, storage = self.get_storage_data()
        if type == '阿里OSS':
            url = upload2ali(storage, filename)
            return url


    def get_storage_data(self):
        self.db.execute(select_sql)
        data = self.db.fetchone()
        bucket_name = data[1]
        extra = data[2]
        key = data[3]
        secret = data[4]
        if data[0] == '阿里OSS':
            bucket = init_ali(bucket_name, extra, key, secret)
            return '阿里OSS', {'bucket': bucket, 'bucket_name': bucket_name, 'extra': extra}

pyperclip.copy(urls['markdown_url'])