【第5回】pythonで某Authenticator的なツールを作ってみた

少し失敗している（ちゃんと動作しない）

• あえて失敗しているのも載せます。
• 今回は、メインの動作部分をきれいにしました。
• 上部のメニューから新規登録の部分を動作させるために改造してた。
• メニュー＞閉じるはできるようになった。

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import tkinter as tk
import pyotp
import sqlite3
import os
import pyautogui
import sys

def create_table():
    # データベースに接続する
    conn = sqlite3.connect('gauth.db')
    c = conn.cursor()
    # テーブルの作成
    sql='''CREATE TABLE gauth
    (id integer primary key AUTOINCREMENT,
     name text,
     private_key text)'''

    c.execute(sql)
    # 挿入した結果を保存（コミット）する
    conn.commit()
    # データベースへのアクセスが終わったら close する
    conn.close()

def view():
    # データベースに接続する
    conn = sqlite3.connect('gauth.db')
    c = conn.cursor()
    for a in c.execute("select * from gauth"):
        totp = pyotp.TOTP(a[2])  # 鍵的な値
        totp.now()
        # 表示用ラベル
        Static1 = tk.Label(text=a[1])
        Static1.pack(side='left')

        Static2 = tk.Label(text=totp.now())
        Static2.pack(side='left')
    # 挿入した結果を保存（コミット）する
    conn.commit()
    # データベースへのアクセスが終わったら close する
    conn.close()

def insert(id,username,private_key):
    # データベースに接続する
    conn = sqlite3.connect('gauth.db')
    c = conn.cursor()
    # データの挿入
    c.execute(
        "INSERT INTO gauth VALUES (?,?,?)",(id,username,private_key))
    # 挿入した結果を保存（コミット）する
    conn.commit()
    # データベースへのアクセスが終わったら close する
    conn.close()

def window_menu():
    # メニューバー
    menubar = tk.Menu(root)
    filemenu = tk.Menu(menubar)
    filemenu.add_command(label="新規登録", command=new)
    filemenu.add_command(label="登録削除")
    filemenu.add_command(label="閉じる", command=close)
    menubar.add_cascade(label="ファイル", menu=filemenu)
    root.config(menu=menubar)

def new():
    pyautogui.prompt(text='user_name', title='user_name' , default='')
    pyautogui.prompt(text='private_key', title='private_key' , default='')
    #insert(user_name,private_key)

def close():
    sys.exit()

# tkinterでwindowの作成とタイトルを作る
# windowサイズの指定
root = tk.Tk()
root.title(u"g_authentication_tool")
root.geometry("300x200")
window_menu()
if(os.path.exists('gauth.db')):
    view()
else:
    create_table()
    #insert(1,"user1",'base32secret3232')
    view()
root.mainloop()