Python 到 Go 语言的转换：证书下载和配置文件写入

Python 到 Go 语言的转换：证书下载和配置文件写入

本文将展示如何将一个 Python 脚本转换为 Go 语言代码，实现证书下载和配置文件写入的功能。原始 Python 代码使用 tkinter 库创建了一个简单的 GUI 界面，用于输入编号并下载证书。

Python 代码

import tkinter as tk
from tkinter import messagebox
import os
import configparser
import requests
import sys


# 下载证书
def download_file():

    proxies = {
        "http": "http://127.0.0.1:8080",
        "https": "http://127.0.0.1:8080    ",
    }
    url = "http://mitm.it/cert/p12"
    # 下载到桌面
    filename = os.path.join(os.path.expanduser("~\Desktop"), 'mitmproxy-ca-cert.p12')
    # 发送请求并下载文件
    response = requests.get(url=url, proxies=proxies)
    with open(filename, "wb") as f:
        f.write(response.content)


# 写入配置文件
def set_ini(name):
    path = os.path.join(os.environ['USERPROFILE'], 'HC.ini')  # 获取文件路径
    config = configparser.ConfigParser()
    config.add_section('general')
    with open(path, mode='w') as config_file:
        config.write(config_file)


class App:



    def button_click(self):
        entry_text = self.entry.get()  # 获取编辑框内容        else:

        if entry_text == '':
            messagebox.showerror('错误', '请输入编号')
            download_file()
            # cert = '321'
            set_ini(entry_text)
            self.entry.config(state="readonly")
            messagebox.showinfo('信息', '设置完成,证书已下载到桌面请手动安装')
            self.master.destroy()

    def center_window(self):
        '''将窗口居中'''
        width = self.master.winfo_reqwidth()
        height = self.master.winfo_reqheight() // 2
        screen_width = self.master.winfo_screenwidth()
        screen_height = self.master.winfo_screenheight()
        x = int((screen_width - width) / 2)
        y = int((screen_height - height) / 2)
        self.master.geometry("{}x{}+{}+{}".format(width, height, x, y))


a = App()

Go 语言代码

package main

import (
	"fmt"
	"github.com/gen2brain/dlgs"
	"gopkg.in/ini.v1"
	"io/ioutil"
	"net/http"
	"net/url"
	"os"
	"path/filepath"
)

type App struct {
	entry  string
	master string
}

// 下载证书
func download_file() {
	proxyUrl, _ := url.Parse("http://127.0.0.1:8080")
	transport := &http.Transport{Proxy: http.ProxyURL(proxyUrl)}
	client := &http.Client{Transport: transport}
	url := "http://mitm.it/cert/p12"
	// 下载到桌面
	home, _ := os.UserHomeDir()
	filename := filepath.Join(home, "Desktop", "mitmproxy-ca-cert.p12")
	// 发送请求并下载文件
	response, _ := client.Get(url)
	defer response.Body.Close()
	body, _ := ioutil.ReadAll(response.Body)
	ioutil.WriteFile(filename, body, 0644)
}

// 写入配置文件
func set_ini(name string) {
	home, _ := os.UserHomeDir()
	path := filepath.Join(home, "HC.ini") // 获取文件路径
	cfg := ini.Empty()
	cfg.Section("general").Key("name").SetValue(name)
	cfg.SaveTo(path)
}

func (a *App) button_click() {
	entry_text := a.entry // 获取编辑框内容
	if entry_text == "" {
		dlgs.Error("错误", "请输入编号")
		download_file()
		// cert := "321"
		set_ini(entry_text)
		dlgs.Info("信息", "设置完成,证书已下载到桌面请手动安装")
		os.Exit(0)
	}
}

func (a *App) center_window() {
	// 将窗口居中
	width, height := 300, 200
	screen_width, screen_height := 1024, 768
	x := int((screen_width - width) / 2)
	y := int((screen_height - height) / 2)
	fmt.Printf("%dx%d+%d+%d\n", width, height, x, y)
}

func main() {
	a := &App{}
	a.entry, _ = dlgs.Entry("请输入编号", "请输入编号", "")
	a.button_click()
}

总结

本文展示了将 Python 脚本转换为 Go 语言代码的示例，并解释了代码的逻辑和实现细节。Go 语言的代码结构更加简洁，使用 net/http 包和 io/ioutil 包进行网络请求和文件操作。此外，Go 语言的错误处理方式更加规范，使用 _ 来忽略不需要的错误信息。

希望本文能帮助您更好地理解 Python 和 Go 语言之间的转换，并能够根据实际需求进行代码迁移。