예제 #1
0
 def test_delete_project(self):
     Projects.Delete_project(self.con, 13)
     cursorObj = self.con.cursor()
     cursorObj.execute('Select * from Projects where Projectid=?', (13, ))
     output = cursorObj.fetchone()
     self.assertIsNone(output)
예제 #2
0
    def BDO(buname):
        while True:
            print(
                "\nMenu\n(1)Create GPM accounts\n(2)Delete GPM accounts\n(3)Update GPM\n(4)Create Project\n(5)Update "
                "Project\n(6)Delete Project\n(7)Show GPM accounts\n(8)Show Projects\n(9)Show member accounts\n(10)See "
                "the complaints\n(11)Approve Pending Requests\n(12)Previous Menu"
            )
            choice = input(">>> ").lower().rstrip()
            if choice == "1":
                print('Here you can add new GPM accounts')
                bdoid = buname

                User.sign_up(con, 'gpm', bdoid)

            elif choice == "2":
                print('Here you can delete GPM accounts')
                gpms = Grama.List_of_gpms(con, buname)
                pd.set_option('display.width', None)
                df = pd.DataFrame(gpms)
                df.columns = [
                    'name', 'username', 'area', 'pincode', 'BdoUsername'
                ]
                print(df)
                uname = input("Please enter a username to be deleted : ")
                User.delete_account(con, uname, 'gpm')

            elif choice == "3":
                print('Here you can Update GPM accounts')
                gpms = Grama.List_of_gpms(con, buname)
                pd.set_option('display.width', None)
                df = pd.DataFrame(gpms)
                df.columns = [
                    'name', 'username', 'area', 'pincode', 'BdoUsername'
                ]
                print(df)
                gpmid = input('Enter the GpmId you want to update : ')
                Grama.Update_GPM(con, buname, gpmid)

            elif choice == "4":
                print('Here you can Create a project')
                Projects.Create_project(con)

            elif choice == "5":
                cursorObj = con.cursor()

                print('Here you can Update a ProjectMain')
                Projects.show_projects(con)
                projid = input(
                    'Enter the id of the project you want to edit : ')
                cursorObj.execute('SELECT * from Projects where Projectid=?',
                                  (projid, ))
                projects = cursorObj.fetchone()
                if projects is not None:
                    Projects.Update_project(con, projid)
                else:
                    print('No project found with the id', projid)

            elif choice == "6":
                print('Here you can Delete a project')
                Projects.show_projects(con)
                projid = input(
                    'Enter the id of the project you want to delete : ')
                Projects.Delete_project(con, projid)

            elif choice == "7":
                print('Here you can see all the accounts')
                gpms = Grama.List_of_gpms(con, buname)
                pd.set_option('display.width', None)
                df = pd.DataFrame(gpms)
                df.columns = [
                    'name', 'username', 'area', 'pincode', 'BdoUsername'
                ]
                print(df)

            elif choice == "8":
                print('Here you can see all the projects')
                Projects.show_projects(con)
            elif choice == "9":
                print('Here you can see all the member accounts')
                cursorObj = con.cursor()
                cursorObj.execute(
                    'SELECT Members.username,Members.Gpmapproved,Members.Bdoapproved,Members.WorkingDays,'
                    'Members.pincode,Members.wage,Members.attendance,Members.projectid from GPM INNER JOIN '
                    'Members on GPM.username=Members.Gpmid '
                    'where GPM.Bdo_id=?', (buname, ))
                member_under_bdos = cursorObj.fetchall()
                pd.set_option('display.width', None)
                df = pd.DataFrame(member_under_bdos)
                df.columns = [
                    'Username', 'Gpm_Approved', 'Bdo_Approved', 'WorkingDays',
                    'Pincode', 'Wage', 'Attendance', 'ProjectId'
                ]
                print(df)

            elif choice == "10":
                print(
                    'Here you can see all the issues raised by the member under you'
                )
                cursorObj = con.cursor()
                cursorObj.execute(
                    'SELECT Members.username from GPM INNER JOIN Members on GPM.username=Members.Gpmid '
                    'where GPM.Bdo_id=?', (buname, ))
                memberid = cursorObj.fetchall()
                for member in memberid:
                    cursorObj.execute(
                        'SELECT * from Complaints where memberid= ? ',
                        (member[0], ))
                    Complaints_received = cursorObj.fetchall()
                    pd.set_option('display.width', None)
                    df = pd.DataFrame(Complaints_received)
                    try:
                        df.columns = [
                            'Id', 'Issue', 'MemberId', 'Bdo_remarks',
                            'Gpm_remarks'
                        ]
                        print(df)
                    except:
                        print('')
                        main.BDO(buname)
            elif choice == "11":
                Projects.show_projects(con)
                cursorObj = con.cursor()
                cursorObj.execute(
                    'SELECT Members.username,Members.Gpmapproved,Members.Bdoapproved,Members.WorkingDays,'
                    'Members.pincode,Members.wage,Members.attendance,Members.projectid from GPM INNER JOIN '
                    'Members on GPM.username=Members.Gpmid '
                    'where GPM.Bdo_id=?', (buname, ))
                member_under_bdos = cursorObj.fetchall()
                pd.set_option('display.width', None)
                df = pd.DataFrame(member_under_bdos)
                df.columns = [
                    'Username', 'Gpm_Approved', 'Bdo_Approved', 'WorkingDays',
                    'Pincode', 'Wage', 'Attendance', 'ProjectId'
                ]
                print(df)
                memuname = input(
                    'Enter the member username you want to approve : ')
                Projects.Approve_Projects(con, memuname)

            elif choice == "12":
                main.Login_master(con)

            else:
                print("Invalid choice, please choose again\n")