用PySimpleGui做户籍资料查询工具

前言

最近手里有一张几千人的户籍信息表格，突发奇想，想用python做一个小工具，可以按照人名搜索，获取该人所在户的所有人信息，然后显示图片，个人详情页，方便查询
以前用tkinter做过一个从xls表格获取行信息填入doc，并批量打印的小工具，tkinter是比较好用，可是想做好看真心不容易，搜索之下发现了PySimpleGui，这个更加简单
学习python属于个人兴趣，没有接受系统学习，风格也是野路子，写下这篇心得纯粹是记录学习内容，毕竟功能实现都是自己一点点想的，没有参考别人的案例。同为初学者的可以看看，抛砖引玉。大神也请赐教。

逻辑部分

数据源

我手里有一份整村的村民户籍表，不是很完整，错误也比较多，没有身份证号，所以没有泄密嫌疑。表格是xls的，为了更好操作，先手动转成csv格式，当数据库用了。表格内容较多，我们主要用到的是姓名、性别、户主关系、电话号码，几项

需求分析

我想要实现的一个主要功能是，输入一个人名，在数据表中搜索到所有这个名字的人（当然有重名的），然后根据他的户主关系，找到他整户的人，并打印出来。这样有几个技术问题需要解决。。。（我本人纯属爱好，没有接触过算法，都是用最笨的方法解决）

思路及方法

获取csv所有行的内容

这是csv基础操作了，上代码

def open_csv():with open('sunzhuang.csv', 'r') as csvfile:reader = csv.DictReader(csvfile)return reader

根据每行里“户主关系”内容，找到一共有多少户主，并将所有户主的序号记录

这个就是突发奇想了，我找到所有的户主，并把它们的序号单独存一个list，这样确定了被搜索的人的序号，就可以通过序号，确定他在哪两位“户主”之间，从而找到所有的家庭成员。ok，上代码

# 获取所有属性是户主的村民序号 存到huzhu_list列表并返回
def get_huzhu_list():with open('sunzhuang.csv', 'r') as csvfile:reader = csv.DictReader(csvfile)for row in reader:if row['与户主关系'] == '户主':# print(row)huzhu_list.append(row['序号'])return huzhu_list

找到被搜索人在csv中的序号

# 根据姓名 获取她的序号
# 考虑到会有重名的情况，所以用一个列表name_id保存了所有了该名字的序号
def get_name_id(name):name_id = []with open('sunzhuang.csv', 'r') as csvfile:reader = csv.DictReader(csvfile)for row in reader:if row['姓名'] == name:name_id.append(row['序号'])# print('序号是：' + name_id)return name_id

接下来，就是找到被搜索人全家了

# 这是核心了，通过用户序号 找到该户的户主和下一户的户主序号，从而计算出该户所有人的序号
def get_hole_family_id(huzhu_list, name_id):first = 0last = 0for i in huzhu_list:# 首先，huzhu_list里边是所有户主的id# 如果搜索的人就是户主，那么他的id就是该户的第一个人，# 先用huzhu_list.index(i)找到他在户主列表里的位置，+1 找到下一个户主的id# 然后在通过这个位置，找到下一个户主的id 说起来有点绕# 这里只是定位到下一个户主的位置，而不用-1 因为下边用到的# range（first，last）函数构建列表时正好是 first到last-1if int(i) == int(name_id):first = int(i)last = int(huzhu_list[huzhu_list.index(i) + 1])break# 如果要搜索的人不是户主，那么先找他的下一个户主的位置，再通过-1找到本户主的位置elif int(i) > int(name_id):last = int(huzhu_list[huzhu_list.index(i)])first = int(huzhu_list[huzhu_list.index(i) - 1])breakfamily_id_list = [i for i in range(first, last)]return family_id_list

这样，我们得到了被搜索人全家的人员序号

再用两个方法，根据序号，获取csv文件里的整行信息

# 根据序号 定位到csv文件中的行
def get_content(name_id):with open('sunzhuang.csv', 'r') as csvfile:reader = csv.DictReader(csvfile)for row in reader:if row['序号'] == name_id:print(row)

# 通过全家人的序号，获取全家人的信息，并全部添加到列表all_familly_list中
def get_hole_family_cont(family_id_list):all_familly_list = []with open('sunzhuang.csv', 'r') as csvfile:reader = csv.DictReader(csvfile)for row in reader:for id in family_id_list:if row['序号'] == str(id):# row 是一个字典格式，需要用dict.values()将键值转化为列表list3 = list(row.values())# 从列表中找出我们需要的信息，从新创建一个列表，就得到了该人的信息family_list = [list3[3], list3[4], list3[7], list3[6]]all_familly_list.append(family_list)return all_familly_list

至此，通过一个人名，我们得到了他家所有人的信息，并全部装入一个列表里。下面就是做一个页面，把他们显示出来

界面部分

PySimpleGui介绍

网上介绍很多，我摘抄了一点过来，实现很简单，布局采用layout行式布局，很直观
1.import 库
2.layout UI布局
3.window 窗口显示
4.Event loop 事件循环，用户持续交互
5.close 关闭窗口

# 1 - The import
import PySimpleGUI as sg
# 2 - Layout definition
layout = [[sg.Text('My layout')],[sg.Input(key='-INPUT-')],[sg.Button('OK'), sg.Button('Cancel')] ]
# 3 - Create window
window = sg.Window('Design Pattern 3 - Persistent Window', layout)
# 4 - Event Loop
while True:event, values = window.read()if event in (None, 'Cancel'):break# 5 - Close window
window.close()

效果是

基本很容易实现我想要的，其他不再累述，需要的可以去csdn找，很多很多

界面布局

通过修改一个案例，一番折腾，得到了我想要的布局如下

代码如下：

import PySimpleGUI as sg
from PySimpleGUI import InputCombo, Combo, Multiline, ML, MLine, Checkbox, CB, Check, Button, B, Btn, ButtonMenu, \Canvas, Column, Col, Combo, Frame, Graph, Image, InputText, Input, In, Listbox, LBox, Menu, Multiline, ML, MLine, \OptionMenu, Output, Pane, ProgressBar, Radio, Slider, Spin, StatusBar, Tab, TabGroup, Table, Text, Txt, T, Tree, \TreeData, VerticalSeparator, Window, Sizersg.theme('GreenTan')
list = [['    ','  ','户主','      ']]col3 = Column([[Frame('Information:', [[Text(), Column([[Text('户      数:  ')],[Text('人      数:  ')],[Text('低保人数:  ')],[Text('五保人数:  ')],],size=(210, 120), pad=(0, 0))]])]], pad=(0, 0))col2 = Column([[Frame('Accounts:', [[Column([[Table(values=list,headings=['  姓  名  ',' 性  别 ',' 属  性 ','   电     话   '],max_col_width=300,auto_size_columns=True,#自动调整列宽（根据上面第一次的values默认值为准，update时不会调整）display_row_numbers=True,#序号justification='center',#字符排列 left right centernum_rows=12,#行数row_height=30,#行高key='_table_',font=('微软雅黑', 12),text_color='black',background_color='white',enable_events=True,bind_return_key=True,tooltip='This is a table')], ], size=(450, 400),pad=(0, 0))]])]],pad=(0, 0))col1 = Column([# Information frame[Frame('Search:', [[Text(), Column([[Text('输入姓名:')],[Input(key='-USERID-IN-', size=(16, 1)),Button('Search', key='-SEAR-')],], size=(195, 120), pad=(0, 0))]])], ], pad=(0, 0))layout = [[col1,col3], [col2]]window = Window('人员信息查询', layout)while True:event, values = window.read()print(event, values)if event == sg.WIN_CLOSED:breakelif event == '-SEAR-':#print(values['-LIST-'][0])sg.popup('This is {}'.format(values['_table_'][0]))window.close()

现在逻辑部分和界面都有了，把他们融合

这一点颇费脑筋，无奈基础太差，暂时不用面向对象了
基本方法是通过按钮事件的触发，调用逻辑代码，实现功能，并通过更新窗口，使结果显示出来

小技巧

不算技巧吧，是PySimpleGui一个比较方便的地方，每个窗口组件都有一个key属性和一个value属性（text没有），key是组件的名字，value是他要显示的内容，通过window[‘key’]，可以定位该组件，通过window[‘key’].update(‘value’),可以直接将value更新到组件的value属性

最终界面

完整代码


import csv
import PySimpleGUI as sg
from PySimpleGUI import InputCombo, Combo, Multiline, ML, MLine, Checkbox, CB, Check, Button, B, Btn, ButtonMenu, \Canvas, Column, Col, Combo, Frame, Graph, Image, InputText, Input, In, Listbox, LBox, Menu, Multiline, ML, MLine, \OptionMenu, Output, Pane, ProgressBar, Radio, Slider, Spin, StatusBar, Tab, TabGroup, Table, Text, Txt, T, Tree, \TreeData, VerticalSeparator, Window, Sizernum = 0
# 搜索所有的户主 并将他们的序号填入列表 用来调取整户信息
huzhu_list = []# 通过DictReader方法打开csv文件，可以做到搜索功能
def open_csv():with open('sunzhuang.csv', 'r') as csvfile:reader = csv.DictReader(csvfile)return reader# 获得总户数
def get_familly_num():familly_num = 0with open('sunzhuang.csv', 'r') as csvfile:reader = csv.DictReader(csvfile)for row in reader:if row['与户主关系'] == '户主':familly_num += 1# print(row)return familly_num# 获得总人数
def get_people_num():people_num = 0with open('sunzhuang.csv', 'r') as csvfile:reader = csv.DictReader(csvfile)for row in reader:people_num += 1# print(row)return people_num# 获取所有属性是户主的村民序号 存到huzhu_list列表并返回
def get_huzhu_list():with open('sunzhuang.csv', 'r') as csvfile:reader = csv.DictReader(csvfile)for row in reader:if row['与户主关系'] == '户主':# print(row)huzhu_list.append(row['序号'])return huzhu_list# 根据姓名 获取她的序号
# 考虑到会有重名的情况，所以用一个列表name_id保存了所有了该名字的序号
def get_name_id(name):name_id = []with open('sunzhuang.csv', 'r') as csvfile:reader = csv.DictReader(csvfile)for row in reader:if row['姓名'] == name:name_id.append(row['序号'])# print('序号是：' + name_id)return name_id# 根据序号 定位到csv文件中的行
def get_content(name_id):with open('sunzhuang.csv', 'r') as csvfile:reader = csv.DictReader(csvfile)for row in reader:if row['序号'] == name_id:print(row)# 这是核心了，通过用户序号 找到该户的户主和下一户的户主序号，从而计算出该户所有人的序号
def get_hole_family_id(huzhu_list, name_id):first = 0last = 0for i in huzhu_list:# 首先，huzhu_list里边是所有户主的id# 如果搜索的人就是户主，那么他的id就是该户的第一个人，# 先用huzhu_list.index(i)找到他在户主列表里的位置，+1 找到下一个户主的id# 然后在通过这个位置，找到下一个户主的id 说起来有点绕# 这里只是定位到下一个户主的位置，而不用-1 因为下边用到的# range（first，last）函数构建列表时正好是 first到last-1if int(i) == int(name_id):first = int(i)last = int(huzhu_list[huzhu_list.index(i) + 1])break# 如果要搜索的人不是户主，那么先找他的下一个户主的位置，再通过-1找到本户主的位置elif int(i) > int(name_id):last = int(huzhu_list[huzhu_list.index(i)])first = int(huzhu_list[huzhu_list.index(i) - 1])breakfamily_id_list = [i for i in range(first, last)]return family_id_list# 通过全家人的序号，获取全家人的信息，并全部添加到列表all_familly_list中
def get_hole_family_cont(family_id_list):all_familly_list = []with open('sunzhuang.csv', 'r') as csvfile:reader = csv.DictReader(csvfile)for row in reader:for id in family_id_list:if row['序号'] == str(id):# row 是一个字典格式，需要用dict.values()将键值转化为列表list3 = list(row.values())# 从列表中找出我们需要的信息，从新创建一个列表，就得到了该人的信息family_list = [list3[3], list3[4], list3[7], list3[6]]all_familly_list.append(family_list)return all_familly_list# print(first)# print(huzhu_list)sg.theme('GreenTan')col3 = Column([[Frame('Information:', [[Text(), Column([[Text('户      数:  '), Input(key='-hushu-', size=(10, 1))],[Text('人      数:  '), Input(key='-renshu-', size=(10, 1))],[Text('低保人数:  '), Input(key='-dibao-', size=(10, 1))],[Text('五保人数:  '), Input(key='-wubao-', size=(10, 1))], ],size=(210, 120), pad=(0, 0))]])]], pad=(0, 0))col2 = Column([[Frame('Accounts:', [[Column([[Table(values=[['', '', '', '']],headings=['  姓  名  ', ' 性  别 ', ' 属  性 ', '     身 份 证 号     '],max_col_width=300,auto_size_columns=True,  # 自动调整列宽（根据上面第一次的values默认值为准，update时不会调整）display_row_numbers=True,  # 序号justification='center',  # 字符排列 left right centernum_rows=12,  # 行数row_height=30,  # 行高key='_table_',font=('微软雅黑', 12),text_color='black',background_color='white',enable_events=True,bind_return_key=True,tooltip='This is a table')], ], size=(450, 400), pad=(0, 0))]])]], pad=(0, 0))col1 = Column([# Information frame[Frame('Search:', [[Text(), Column([[Text('输入姓名:')],[Input(key='-USERID-IN-', size=(15, 1)),Button('Search', key='-SEAR-')],], size=(195, 120), pad=(0, 0))]])], ], pad=(0, 0))layout = [[col1, col3], [col2]]window = Window('人员信息查询', layout)while True:event, values = window.read()print(event, values)familly_num = get_familly_num()people_num = get_people_num()window['-hushu-'].update(familly_num)window['-renshu-'].update(people_num)if event == sg.WIN_CLOSED:breakelif event == '-SEAR-':# print(values['-LIST-']])if values['-USERID-IN-'] != '':name = values['-USERID-IN-']else:sg.popup('输入一个人名，比如：张三')name = '张三'id_list = get_name_id(name)list1 = get_huzhu_list()all_list = []num = 0for id in id_list:family_id_list = get_hole_family_id(list1, id)familly_list = get_hole_family_cont(family_id_list)num += 1for i in familly_list:all_list.append(i)window['_table_'].Update(all_list)sg.popup('一共找到{}户'.format(num))window.close()

总结

写这个小东西，从构思，查资料，写代码，查bug，用了一整天的时间，对我一个小白爱好者来说还是挺有成就感的。后续将会继续添加照片显示，家庭成员详细信息页面，和添加、删除、修改功能，实现了再分享。