def decorated(*args, **kwargs):
token = get_token()
try:
database = Database()
data = jwt.decode(
token,
'trulysKey',
)
query = database.get_user_by_value('users', 'user_id',
data['user_id'])
if not query:
return {"message": "User does not exist"}, 400
current_user = User(query[0], query[1], query[2], query[3],
query[4], query[5])
except jwt.ExpiredSignatureError as e:
return response_message('Error',
'Signature expired,please login again',
401)
except jwt.InvalidSignatureError as serr:
return response_message('Error',
'Signature is invalid,please login again',
401)
except jwt.DecodeError:
return response_message('Error', 'please login', 401)
return f(current_user, *args, **kwargs)
def __init__(self, master, **options):
super().__init__(master)
self.options = options
self.database = Database()
self.plan_exists_in_db = False
self.master = master
self.bind('<Control-s>', self.save)
self.create_widgets(**options)
def __init__(self):
self.base_price = 1.5
self.trulyKey = os.environ.get('TRULYS_KEY', '')
self.categories = {
"0-5": 1,
"6-30": 3,
"31-60": 6,
"70-100": 9,
"101-300": 12,
"301-499": 15,
"500-1000": 18,
"1001-5000": 25
}
db = Database()
self.parcels = db.get_all_parcels()
def get_users(current_user):
if not db.is_admin(current_user.user_id):
return response_message('unauthorized operation',
'Only admin users can view all users', 401)
users = Database().get_users()
user_list = []
for user in users:
user_dict = {
"user_id": user[0],
"fullname": user[1],
"username": user[2],
"email": user[3],
"phone_number": user[5],
"is_admin": user[6],
"joined": user[7]
}
user_list.append(user_dict)
return jsonify({"users": user_list}), 200
def check_user(self):
db = Database()
if db.check_table("users"):
query = "insert into users(username, email, password) values (%s, %s, %s)"
user_details = (self.username, self.email,
sha256.hash(self.password))
db.add_user(self.username, self.email, sha256.hash(self.password))
return ("User added")
else:
db.create_connection().execute(
"create table users(user_id serial primary key,username varchar(25) not null, email varchar(25) not null, password varchar not null)"
)
db.create_connection().close
db.commit()
query = "insert into users(username, email, password) values (%s, %s, %s)"
user_details = (self.username, self.email, self.password)
db.create_connection().execute(query, user_details)
db.create_connection().con.commit()
return ("User created")
def __init__(self, launcher, name, shard_ids, max_shards):
self.launcher = launcher
self.process = None
self.kwargs = dict(
intents=INTENTS,
allowed_mentions=NO_MENTIONS,
case_insensitive=True,
token=TOKEN,
shard_ids=shard_ids,
shard_count=max_shards,
cluster_name=name,
cache=Cache(),
db=Database(
os.getenv("DB_NAME"),
os.getenv("DB_USER"),
os.getenv("DB_PASSWORD"),
),
theme_color=config.THEME_COLOR,
dark_theme_color=config.DARK_THEME_COLOR,
error_color=config.ERROR_COLOR,
initial_extensions=EXTENSIONS,
)
self.name = name
self.log = logging.getLogger(f"Cluster#{name}")
self.log.setLevel(logging.DEBUG)
hdlr = logging.StreamHandler()
hdlr.setFormatter(
logging.Formatter(
"[%(asctime)s %(name)s/%(levelname)s] %(message)s"))
fhdlr = logging.FileHandler("logs/cluster-Launcher.log",
encoding="utf-8")
fhdlr.setFormatter(
logging.Formatter(
"[%(asctime)s %(name)s/%(levelname)s] %(message)s"))
self.log.handlers = [hdlr, fhdlr]
self.log.info(f"Initialized with shard ids {shard_ids}, "
f"total shards {max_shards}")
import datetime
from app.auth.decorator import response_message, token_required
from werkzeug.security import generate_password_hash, check_password_hash
from flask import Blueprint, request, jsonify
from flasgger import swag_from
from validate_email import validate_email
from app.database.database import Database
from app.model.models import User
from app.views.parcels import sendemail
import os
from app.auth.decorator import get_token
from flask_cors import CORS
auth = Blueprint('auth', __name__)
db = Database()
CORS(auth)
@auth.route('/api/v2/auth/signup', methods=['POST'])
@swag_from('../doc/signup.yml')
def create_user():
"""
User creates an account
User sign up details are added to the data base
"""
if request.content_type != 'application/json':
return response_message('Bad request',
'Content-type must be json type', 400)
request_data = request.get_json()
try:
#! /usr/bin/env python3
"""Script for Database creation.
Force to remove database"""
from app.database.database import Database
import config
db = Database(config.DBCONNECT)
"""CREATE AND FILL IN"""
db.cursor = db.mydb.cursor()
db.cursor.execute("DROP DATABASE IF EXISTS {};".format(config.DBCONNECT["DATABASE"]))
db.cursor.close()
db.create_database()
db.fill_in_database()
def tearDown(self):
db = Database()
db.drop_tables()
def test_db_connection(self):
db = Database()
self.assertTrue(db.connection)
def test_db(self):
db = Database()
self.assertTrue(db)
def setUp(self):
self.app = app.test_client()
db = Database()
db.create_tables()
def setUpModule():
global db
db = Database(os.environ["DBURI"], True)
def run_languages() -> None:
github = GithubScraper()
database = Database()
data = github.get_languages()
database.save_languages(languages=data)
def run_commits() -> None:
github = GithubScraper()
database = Database()
data_days, data_hours = github.get_commits()
database.save_commits(days=data_days, hours=data_hours)
