def init():
log.warning( f'running from {os.getcwd()}' )
if not os.path.isfile(db.db_file):
log.warning("No database file found at {}".format(db.db_file))
h.printcolor("No database file found at {}".format(db.db_file),
color = h.bcolors.WARNING)
db.create_database()
else:
log.warning("database file OK")
h.printcolor("database file OK",
color = h.bcolors.OKGREEN)
db.connect_database()
db.init()
def test_save_sorted_users(self):
cur, conn, dbname = db.connect_database()
db.del_all_photos(cur, conn)
cur.execute('select count(*) from Person')
result = cur.fetchall()
self.assertEqual(result[0][0], 0)
self.assertEqual(len(self.user.sorted_photos), 10)
for k, v in self.user.sorted_photos.items():
like_count = 0
for item in v:
if isinstance(item, type(dict)):
item_like = item.get('likes')
if item_like and like_count == 0:
self.assertGreater(item_like, like_count)
elif item_like:
self.assertGreater(like_count, item_like)
like_count = item_like
self.user.save_photos_db()
cur.execute('select count(*) from Person')
result = cur.fetchall()
self.assertEqual(result[0][0], 10)
current_path = str(os.path.dirname(os.path.abspath(__file__)))
current_path = current_path[:-4]
save_directory = os.path.join(current_path, 'fixtures\\test_data')
self.user.save_photos_json(save_directory)
save_directory += '.json'
with open(save_directory, encoding='utf-8') as new_directory:
test_data = json.load(new_directory)
self.assertEqual(len(test_data), 10)
def users_from_db(self):
cur, conn, dbname = db.connect_database()
result = db.get_all_ids(cur)
ids = []
for item in result:
ids.append(item[0])
print(ids)
def auth_register():
user = request.get_json()
conn = db.connect_database()
if db.add_user(conn, user):
return 'user register success'
else:
return 'user register fail'
def note_list():
access = check_authorization(request)
if access['valid']:
conn = db.connect_database()
notes = db.get_all(conn)
return jsonify(notes), 200
else:
response = {'message': access['message']}
return jsonify(response), 401
def note_add():
access = check_authorization(request)
if access['valid']:
note = request.get_json()
conn = db.connect_database()
if db.add_note(conn, note):
response = {'message': 'notes add success'}
return jsonify(response), 200
else:
response = {'message': 'notes add fail'}
return jsonify(response), 500
else:
response = {'message': access['message']}
return jsonify(response), 401
def add_like(self, user_id):
vk = self.vk_session.get_api()
cur, conn, dbname = db.connect_database()
photos = db.get_photos(user_id, cur)
for photo in photos[0][-1]:
if isinstance(photo, dict):
current_liked = self.is_liked(user_id, photo['id'])
if current_liked == 0:
print(f"Do you want to add like to photo {photo['url']}? Input Yes or No: ")
person_input = input().strip()
if person_input.lower() == 'yes':
vk.likes.add(type='photo', owner_id=int(user_id), item_id=int(photo['id']))
print(f"Photo with id {photo['id']} was liked!")
print("All photos have been looked!")
def auth_login():
data = request.get_json()
conn = db.connect_database()
user = db.get_user_by_username(conn, data['username'])
if data['password'] == user['password']:
access_token = generate_token(user['id'])
if access_token:
response = {
'message': 'You logged in successfully.',
'access_token': access_token.decode()
}
return jsonify(response), 200
else:
response = {
'message': 'Invalid username or password, Please try again.'
}
return jsonify(response), 401
def notes_manipulation(nid):
access = check_authorization(request)
if access['valid']:
conn = db.connect_database()
if request.method == 'GET':
note = db.get_note(conn, nid)
if note:
return jsonify(note)
elif request.method == 'PUT':
note = request.get_json()
note_update = db.update_note(conn, nid, note)
return jsonify(note_update)
else:
if db.del_note(conn, nid):
return 'delete success'
response = {'message': 'resource not found'}
return jsonify(response), 404
else:
response = {'message': access['message']}
return jsonify(response), 401
from db import connect_database, init_database
conn = connect_database('admin', 'postgres', '127.0.0.1', '5432', 'flask_api')
init_database(conn)
def connect_db(self):
if not self.is_db_connected():
self.db = connect_database(host, user, password, db)
def initialise(): g.db = connect_database()
slug = html.escape(row[10])
# parse the list of genres
genres = self.parse_list(row[11])
# parse the list of studios
studios = self.parse_list(row[12])
# parse the list of publishers
publishers = self.parse_list(row[13])
# parse the list of franchises
franchises = self.parse_list(row[14])
# parse the list of regions
regions = self.parse_list(row[15])
# check if this item already exists by name
if db.get_one_item(name=name) != None:
continue
item_id = db.insert_item(media_type, name, short_name, long_desc,
short_desc, review_url, review_score,
slug)
db.insert_genre_item_joins(genres, item_id)
db.insert_studio_item_joins(studios, item_id)
db.insert_publisher_item_joins(publishers, item_id)
db.insert_franchise_item_joins(franchises, item_id)
db.insert_region_item_joins(regions, item_id)
if __name__ == '__main__':
mysql_db = db.connect_database()
csv_reader = CSVReader(mysql_db)
csv_reader.load('cfgames.csv')
db.close()
def put_user_to_blacklist(self, user_id):
cur, conn, dbname = db.connect_database()
db.put_id_to_blacklist(user_id, cur, conn)
print(f'Person with id {user_id} now is in blacklist!')
def make_user_chosen(self, user_id):
cur, conn, dbname = db.connect_database()
db.make_id_chosen(user_id, cur, conn)
print(f'Person with id {user_id} now is chosen!')
def find_user_from_db(self, user_id):
cur, conn, dbname = db.connect_database()
result = db.get_person_by_id(user_id, cur)
return result
def save_photos_db(self):
cur, conn, dbname = db.connect_database()
db.create_table(cur, conn)
for k, v in self.sorted_photos.items():
db.add_photos(k, v, cur, conn)
print(f'Photos have been saved in database {dbname}!')
