Gfrktyl

I just finished working on a personal project, it is a sort of email type program where you create an account and you can message people, the code works (as far as I’m aware) but it requires review, different areas have different indents, I add more variable than I need, I print things that don’t need to be printed and probably way more than that.

(I realise that it will say import getpass ,and have nothing for it in the code , that is because i forgot to change the password inputs to getpass inputs before putting it on my drive.)

Where the code is located in your files you need a folder called Data with txt files: accountIDs, addCommandList, isAdmin, names, nicknames, passwords.

A folder called messages in it a folder called received.

import random

from random import *

import datetime

import string

import os

import getpass

global usernames

global passwords

global jointPassList

usernames = [line.strip()for line in open('Data/names.txt', 'r')]

passwords = [line.strip()for line in open('Data/passwords.txt', 'r')]

isAdmin = [line.strip()for line in open('Data/isAdmin.txt', 'r')]

accountIDs = [line.strip()for line in open('Data/accountIDs.txt', 'r')]

nicknames = [line.strip()for line in open('Data/nicknames.txt', 'r')]

jointPassList ='n'.join(map(str, passwords))



def main():

                print('n')

                print('n')

                print('/help (1/2) for help')

                currentUserIndex = usernames.index(ascname)

                command =input('>>>')

                if command =='/help':



                                print('/signout       | sign outn/details        | account detailsn/setpas        | set passwordn/userlist     | all usersn/mymessages | your messagesn/message | messagen/addcommand | add command')

                                input()

                                main()

                if command =='/details':



                                print('Name:      ', ascname)

                                print('AccountID: ', accountIDs[currentUserIndex])

                                print('Nickname:  ', nicknames[currentUserIndex])

                                input()

                                main()

                if command =='/setpas':



                                newpas =input('Enter your new password: ')

                                passwords[currentUserIndex] = newpas

                                jointPassList ='n'.join(map(str, passwords))

                                openfile =open('Data/passwords.txt', 'w')

                                openfile.write(jointPassList)

                                openfile.close()                      

                                input()

                                main()

                if command =='/userlist':

                                userlist =open('Data/names.txt').read()

                                print(userlist)

                                input()

                                main()

                if command =='/message':

                                whatuser =input('What user: ')

                                if whatuser in usernames:

                                    message =input('What message would you like to send: ')

                                    openfile =open('Data/messages/recieved/'+whatuser+'.txt', 'a')

                                    date = str(datetime.datetime.now())

                                    openfile.write(date + ' : ')

                                    openfile.write(message+'n')

                                    openfile.close()

                                    input()

                                    main()



                                elif whatuser not in usernames:

                                                print('Nobody was found.')

                                                input()

                                                main()



                if command =='/mymessages':

                                messagesList = [line.strip()for line in open('Data/messages/recieved/'+ascname+'.txt', 'r')]

                                messages = 'n'.join(messagesList)

                                print(messages)

                                input()

                                main()



                if command =='/addcommand':

                                openfile =open('Data/addCommandList.txt', 'a')

                                addcommand =input('What would you like see added to this database: ')

                                openfile.write(addcommand+'n')

                                openfile.close()

                                input()

                                main()



                if command =='/admin':

                                print(isAdmin[currentUserIndex])                               

                                if isAdmin[currentUserIndex] =='True':

                                                print('Nice :)')

                                                input()

                                                main()



                                elif isAdmin[currentUserIndex] =='False':

                                                print('You are not an Admin')

                                                change =input()

                                                if change =='False':

                                                                isAdmin[currentUserIndex] = True

                                                                main()

                                                else:

                                                                main()



                                if isAdmin[currentUserIndex] =='False':

                                                delete =input('Are you sure you would like to delete your account: ')

                                                if delete =='y':

                                                                accountIDs.remove(accountIDs[currentUserIndex])

                                                                isAdmin.remove(isAdmin[currentUserIndex])

                                                                usernames.remove(usernames[currentUserIndex])

                                                                passwords.remove(passwords[currentUserIndex])

                                                                nicknames.remove(nicknames[currentUserIndex])

                                                                os.remove('Data/messages/recieved/'+ascname+'.txt')

                                                                openfile = open('Data/names.txt', 'w')

                                                                openfile.write(name + 'n')

                                                                openfile.close()



                                                                openfile = open('Data/accountIDs.txt', 'w')

                                                                openfile.write(str(accountID) + 'n')

                                                                openfile.close()



                                                                openfile = open('Data/nicknames.txt', 'w')

                                                                openfile.write(nickname + 'n')

                                                                openfile.close()



                                                                openfile = open('Data/passwords.txt', 'w')

                                                                openfile.write(password + 'n')

                                                                openfile.close()



                                                                openfile = open('Data/isAdmin.txt', 'w')

                                                                openfile.write(adminFalse + 'n')

                                                                openfile.close()

                                                                print('Complete...')

                                                                signin()





                if command =='/signout':



                                areYouSure =input('Are you sure you would like to sign out(y/n): ')

                                if areYouSure =='y':

                                                print('Signing outn.n.n.')

                                                signin()

                                elif areYouSure =='n':

                                                main()

                                else:

                                                main()





                else:

                                print(command, 'is not a command in our library, type /addcommand to request new command.')

                                input()

                                main()



def signin():

                existingAccount =input('Do you have an existing account (y/n): ')

                if existingAccount =='y':

                                global ascname

                                global ascpass

                                ascname =input('Enter your username: ')

                                currentUsername = ascname

                                if ascname in usernames:

                                                userIndex = usernames.index(ascname)

                                                print('Correct username.')

                                                ascpass =input('Enter your password: ')

                                                while ascpass in passwords:

                                                                passcheck = passwords.index(ascpass)

                                                                if userIndex == passcheck:

                                                                                print('welcome back', ascname + '.')

                                                                                main()



                                                                else:

                                                                                wrongPass =input('Incorrect password.')

                                                                                input()

                                                                                signin()

                                                print('Yes')

                                                wrongPass =input('Incorrect password.')

                                                input()

                                                signin()

                                elif ascname not in usernames:

                                                wrongName =input('Incorrect username.')

                                                input()

                                                signin()

                                else:

                                                #debuging

                                                print('Error')

                                                singin()



                elif existingAccount =='n':

                                name =str(input('Enter your name: '))

                                while len(name) == 0:

                                                name =input("You haven't entered anything, try again.")

                                                input()

                                                signin()

                                if name in open('Data/names.txt').read():

                                                name =input('That name already exists.')

                                                input()

                                                signin()

                                usernames.append(name)

                                password =input('Enter your new password: ')

                                while len(password) < 4:

                                                password =input('Your password must be 5 characters long.')

                                                input()

                                                signin()

                                passwords.append(password)



                                nickname =input('Enter your nickname: ')

                                accountID =random()

                                while accountID in accountIDs:

                                                accountID =random()



                                adminFalse = str(False)

                                isAdmin.append(adminFalse)



                                openfile = open('Data/messages/recieved/' +name+ '.txt',  'w+')

                                openfile.write('n')

                                openfile.close()



                                openfile = open('Data/names.txt', 'a')

                                openfile.write(name + 'n')

                                openfile.close()



                                openfile = open('Data/accountIDs.txt', 'a')

                                openfile.write(str(accountID) + 'n')

                                openfile.close()



                                openfile = open('Data/nicknames.txt', 'a')

                                openfile.write(nickname + 'n')

                                openfile.close()



                                openfile = open('Data/passwords.txt', 'a')

                                openfile.write(password + 'n')

                                openfile.close()



                                openfile = open('Data/isAdmin.txt', 'a')

                                openfile.write(adminFalse + 'n')

                                openfile.close()







                                signin()



                else:

                                signin()



signin()

1. 
   

   Ban yourself from recursion. If you're in the function main don't call main. What you want is to have a while True loop and to use continue.

   
   
   

   while True:
   
       ...
   
       if invalid_username:
   
           continue
   
       ...
   
   

   
   



2. json is likely to make your code much easier to use. As you don't have to have 5 lists for five attributes on a user.



3. 
   cmd.Cmd is probably what you want to use. This is as you can put all your functions in their own functions and you can focus on what the code does.


4. Your code has a lot of useless print input and other nonsense statements. Follow he Unix way and remove them.


5. Global variables are bad. Whilst the changes I'll recommend still use them, I've hidden them behind properties on a class. Which encourages safer programming practices.


6. You should keep the database in memory, and ensure it's written to when the program ends. You can do this via try finally.


7. I've partially changed your program to use cmd.Cmd, changing signin to also be one is up to you. I encourage you to look at the docs as the builtin help command is nice.


8. Wrap your main code in an if __name__ == '__main__' guard.

import random

from random import *

import datetime

import cmd

import json

from pprint import pprint



try:

    f = open('Data.json')

except FileNotFoundError:

    with open('Data.json', 'w') as w:

        json.dump({}, w)

    f = open('Data.json')

finally:

    with f:

        database = json.load(f)

        database.setdefault('users', )

        data = database.setdefault('data', {})

        data.setdefault('command_list', )

user = None





def find_user_name(database, name):

    for user in database['users']:

        if user['name'] == name:

            return user

    return None





class Main(cmd.Cmd):

    prompt = '>>> '



    @property

    def user(self):

        return user



    @property

    def database(self):

        return database



    def do_details(self, arg):

        print('Name:      ', self.user['name'])

        print('AccountID: ', self.user['id'])

        print('Nickname:  ', self.user['nickname'])



    def do_setpas(self, arg):

        self.user['password'] = input('Enter your new password: ')



    def do_userlist(self, arg):

        pprint([u['name'] for u in self.database['users']])



    def do_message(self, target_user):

        t_user = find_user_name(self.database, target_user)

        if t_user is None:

            return



        message = input('Message: ')

        t_user['messages'].append({

            'from': self.user['name'],

            'date': str(datetime.datetime.now()),

            'message': message

        })



    def do_mymessages(self, arg):

        pprint(self.user['messages'])



    def do_addcommand(self, arg):

        self.database['data']['command_list'].append(arg)



    def do_admin(self, arg):

        if not self.user['admin'] and arg == 'True':

            self.user['admin'] = True



    def do_delete(self, arg):

        if self.user['admin']:

            return



        delete = input('Are you sure you would like to delete your account: ')

        if delete != 'y':

            return



        i = self.database.index(self.user)

        self.database.pop(i)



        return True



    def do_signout(self, arg):

        return True





def signin():

    global user



    while True:

        existing_account = input('Do you have an existing account (y/n): ')

        if existing_account == 'y':

            username = input('Username: ')

            user = find_user_name(database, username)

            if user is None:

                print('Invalid username')

                continue



            password = input('Password: ')

            if password != user['password']:

                print('Invalid password')

                continue



            Main().cmdloop()

        else:

            username = str(input('Username: '))

            while not username:

                username = input("You haven't entered anything, try again.")



            user = find_user_name(database, username)

            if user is not None:

                print('That name already exists.')

                user = None

                continue



            password = input('Password: ')

            while len(password) < 4:

                print('Your password must be 5 characters long.')

                password = input('Password: ')



            nickname = input('Enter your nickname: ')

            id_ = random()

            while id_ in [u['id'] for u in database['users']]:

                id_ = random()



            user = {

                'name': username,

                'password': password,

                'nickname': nickname,

                'id': id_,

                'messages': ,

                'admin': False

            }

            database['users'].append(user)

            Main().cmdloop()





if __name__ == '__main__':

    try:

        signin()

    finally:

        with open('Data.json', 'w') as f:

            json.dump(database, f)

Example usage:

Do you have an existing account (y/n): n

Username: Peilonrayz

Password: abcde

Enter your nickname: Peilonrayz

>>> help



Documented commands (type help <topic>):

========================================

help



Undocumented commands:

======================

addcommand  delete   message     setpas   userlist

admin       details  mymessages  signout



>>> details

Name:       Peilonrayz

AccountID:  0.5494927696334424

Nickname:   Peilonrayz

>>> admin

>>> admin True

>>> delete

>>> userlist

['Peilonrayz']

>>> signout

Do you have an existing account (y/n):