forked from pchojka/Py_Expense_template
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathuser.py
More file actions
30 lines (25 loc) · 753 Bytes
/
user.py
File metadata and controls
30 lines (25 loc) · 753 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
from PyInquirer import prompt
import csv
user_questions = [
{
"type":"input",
"name":"username",
"message":"New User - Name: ",
}
]
def add_user():
infos = prompt(user_questions)
with open('users.csv', mode='a', newline='') as file:
writer = csv.writer(file)
writer.writerow([infos['username']])
# This function should create a new user, asking for its name
return
def get_users():
users = []
with open('users.csv', mode='r') as file:
csv_reader = csv.reader(file)
for row in csv_reader:
if row:
user = {"username": row[0].strip()} # Assuming the username is in the first column
users.append(user)
return users