快捷導(dǎo)航

Python實現(xiàn)截圖生成符合markdown的鏈接

更新時間：2022年01月27日 15:35:02 作者：Lpepsi

之前是用的是typora來寫的文章，最近typora最近開始收費了，所以就不想用了，于是找到了一個替代品MarkText。本文將介紹如何通過Python實現(xiàn)截圖自動生成符合markdown的鏈接，感興趣的可以了解一下

背景

之前是用的是typora來寫的文章，最近typora最近開始收費了，所以就不想用了，于是找到了一個替代品MarkText，感覺跟typora差不多

整體樣子就像上面，簡約風(fēng)，個人挺喜歡的。唯一的一個問題就是粘貼圖片的時候，圖片只能放在本地，雖然marktext有圖片上傳的功能，但是只支持GitHub的圖床，設(shè)置了過后經(jīng)常會上傳失敗，導(dǎo)致還是存在本地，存在本地的弊端就是文章在轉(zhuǎn)移的需要把圖片都帶上或者復(fù)制到掘金等平臺的時候，圖片會失效。于是打算用python寫個自動生成markdownUrl的程序。

思路

程序的整體思路是，首先需要給界面來設(shè)置存儲的類型，即是選擇存在阿里OSS還是七牛的Kodo還是其他的云存儲，還有設(shè)置key和secret以及不同存儲類型所需要的屬性，接著界面可以顯示復(fù)制的圖片，以及上傳成功后的markdownUrl和httpUrl,界面大體如下

子界面：

主界面：

實現(xiàn)

程序整體選擇用python來實現(xiàn)，因為之前用過QT，所以GUI的框架用的是pyqt5，數(shù)據(jù)庫用的是sqlite3，還有阿里云和七牛的sdk等。

整個界面有一個主窗口和一個子窗口構(gòu)成，主界面在初始化的同時初始化數(shù)據(jù)庫

class ImgFrame(QMainWindow):
? ? def __init__(self):
? ? ? ? super().__init__()
? ? ? ? self.http_url = None
? ? ? ? self.markdown_url = None
? ? ? ? self.clipboard = None
? ? ? ? self.img = None
? ? ? ? # 初始化數(shù)據(jù)庫
? ? ? ? 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'), '設(shè)置', 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('系統(tǒng)')

? ? ? ? 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)
? ? #返回游標(biāo)
? ? return cursor

子窗口的主要作用就是設(shè)置云存儲所需要的各種字段，然后存儲到數(shù)據(jù)庫中

class secondFrame(QWidget):
? ? def __init__(self, db):
? ? ? ? super().__init__()
? ? ? ? # self.init_ui()
? ? ? ? self.db = db
? ? ? ? self.resize(400, 100)
? ? ? ? self.setWindowTitle('存儲設(shè)置')

? ? ? ? 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)

子窗口是通過主窗口的菜單欄的設(shè)置菜單觸發(fā)

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

監(jiān)聽剪貼板的功能通過pyqt中的clipboard來監(jiān)聽剪貼板的實現(xiàn)

    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}

當(dāng)圖片上傳成功后，默認(rèn)會生成markdown的圖片url,然后將這個url設(shè)置到剪貼板中，然后在marktext中只需要粘貼就能貼上圖床的圖片，因為pyqt為了防止死循環(huán)，不允許通過剪貼板設(shè)置內(nèi)容，所以我們可以通過pyperclip來設(shè)置剪貼板