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}!')