def test_show_available_products(self):
"""Test for show_available_products function from database.py"""
client = MongoClient()
db = client['database']
product = db['product']
add_data(product, 'products.csv')
self.assertEqual(4, len(show_available_products(db)))
test_result = show_available_products(db)
self.assertTrue('9736' in test_result)
self.assertTrue('6001' not in test_result)
drop_all(db)
def test_add_data(self):
"""Test for add_data function from database.py"""
client = MongoClient()
db = client['database']
product = db['product']
customer = db['customer']
rentals = db['rentals']
self.assertEqual(0, add_data(product, 'products.csv'))
self.assertEqual(0, add_data(customer, 'customers.csv'))
self.assertEqual(0, add_data(rentals, 'rentals.csv'))
self.assertEqual(['customer', 'rentals', 'product'],
db.list_collection_names())
drop_all(db)
def test_show_rentals(self):
"""Test for show_rentals function from database.py"""
client = MongoClient()
db = client['database']
customer = db['customer']
rentals = db['rentals']
add_data(customer, 'customers.csv')
add_data(rentals, 'rentals.csv')
test_result = show_rentals(db, '6921')
self.assertTrue('AP987' in test_result)
test_result2 = show_rentals(db, '2953')
self.assertTrue('AP990' in test_result2)
test_result3 = show_rentals(db, '0000')
self.assertEqual(None, test_result3)
drop_all(db)
def add_action(self, message: telebot.types.Message, person_id):
if isinstance(message.text, str):
if message.text.lower() == 'все':
self.bot.send_message(
message.from_user.id, 'Замечательно, все внесено в план! '
'Ты всегда можешь посмотреть свой список дел,'
' просто написав мне \'меню\' и нажав на кнопку '
'\'Посмотреть план на день\'')
else:
database.add_data(person_id, 'plan', message.text)
self.bot.send_message(message.from_user.id, "Что-то еще?")
self.bot.register_next_step_handler(message, self.add_action)
else:
self.bot.send_message(message.from_user.id,
"Давай лучше текстом, хорошо?")
self.bot.register_next_step_handler(message, self.add_action)
def googleSearch(words, matches, min_value, max_value, sleep_time):
try:
if matches:
words = words[21:]
query.append(words)
else:
query.append("You didn't say, 'Hey bot search for me' :(")
query.append(words)
res = "Tab opened with " + str(
max_value) + " links, according settings Min : " + str(
min_value) + ", Max : " + str(
max_value) + ", Sleep time : " + str(sleep_time)
links.append(res)
links.append(" ")
if words:
for data in search(words,
tld="co.in",
num=min_value,
stop=max_value,
pause=sleep_time):
links.append(data)
webbrowser.open(data)
date = str(
datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
db_res = database.add_data(date, words, data)
if db_res == 1:
errors.append(" ")
errors.append(
"Voice search assistant could not connect with database."
)
links.append(" ")
links.append("Your search results are successfully completed.")
links.append(" ")
words = ""
except:
errors.append(
"Voice search assistant could not open web browser for search results, try again."
)
errors.append(" ")
errors.append("Turn ON, to search your queries.")
errors.append(" ")
print('Done from googleSearch')
def menu():
connection = database.connect()
database.create_database(connection)
while (user_input := input(menu_prompt)) != "7":
if user_input == "1":
app = input("Type the name of the app or website: ")
user = input("Type the user name: ")
pw_choice = input('''Please, choose one of these options
1. Type my own password
2. Use the 'Secure Password Generator
Your choice is: ''')
if pw_choice == "1":
password = input("Type your password: "******"2":
PasswordGenerator.pw_lenght()
psswrd = input(
'''If you want to try a new password type "1".
You can still write your own password or if you like the password, you can copy it and paste it here: '''
)
if psswrd != "1":
password = psswrd
else:
continue
else:
print("Sorry, invalid input")
database.add_data(connection, app, user, password)
elif user_input == "2":
data = database.get_all_data(connection)
for info in data:
print(info)
elif user_input == "3":
app = input("Please, type the app you want to delete: ")
database.delete_info(connection, app)
elif user_input == "4":
update = input("Please, type the app you want to update: ")
new_app = input("The new name will be: ")
new_user = input("The new user name will be: ")
new_pw = input("The new password will be: ")
database.modify_column(connection, new_app, new_user, new_pw,
update)
elif user_input == "5":
username = input("Type the user name you are looking for: ")
find_user = database.get_by_user(connection, username)
for user in find_user:
print(user)
elif user_input == "6":
appname = input("Type the app you are looking for: ")
find_app = database.get_by_app(connection, appname)
for app in find_app:
print(app)
else:
print("Invalid input, please try again.")
keys = temp_list.pop(0)
temp_list = [{
keys[i].lower(): dataConversions[i](thing[i])
for i in range(len(keys))
} for thing in temp_list]
return temp_list
if __name__ == "__main__":
import sys
if not "-" in sys.argv:
print("Need an option: try using -p or -p after the arguments")
sys.exit(0)
if "-p" in sys.argv:
if "app" in sys.argv[1].lower():
print(readApplicantCSV(sys.argv[2]))
else:
print(readDataCSV(sys.argv[2]))
if "-l" in sys.argv:
import database
applicants = readApplicantCSV(sys.argv[2])
data = readDataCSV(sys.argv[3])
idDict = {}
for napplicant in applicants:
num = database.add_applicant(**applicants[napplicant])
idDict[napplicant] = num
for datum in data:
datum["applicant"] = idDict[datum["applicantid"]]
del datum["applicantid"]
database.add_data(**datum)
def new_lesson(self, message: telebot.types.Message):
self.buffer_lesson[2] = message.text
database.add_data(message.chat.id, 'timetable',
' '.join(self.buffer_lesson))
self.bot.send_message(message.chat.id,
'Готово! Вписал новое занятие в расписание:)')
def fonct():
if request.method == 'POST':
Protocol = request.form['Protocol']
key = request.form['key']
add_data(Protocol, key)
return render_template('ex.html')
def save_bookmark(self):
add_data(self.restaurant, self.recipe, self.image)
