Bad reputation

.gitignore

@@ -101,4 +101,13 @@ ENV/
 .mypy_cache/
 
 # pycharm files
-.idea/
+.idea/
+
+# vscode files
+settings.json
+
+# database files
+data/
+
+# static files
+static/images/

Forms/forms.py

@@ -0,0 +1,18 @@
+from flask_wtf import FlaskForm
+from wtforms import StringField, PasswordField, SubmitField, BooleanField
+from wtforms.validators import DataRequired, Email, Length, EqualTo
+
+class RegistrationForm(FlaskForm):
+    username = StringField('Username', validators=[DataRequired(), Length(min=2, max=30)])
+    name = StringField('Name', validators=[DataRequired(), Length(min=2, max=50)])
+    email = StringField('Email', validators=[DataRequired(), Email()])
+    password = PasswordField('Password', validators=[DataRequired()])
+    confirmed_password = PasswordField('Confirm password', validators=[DataRequired(), EqualTo('password')])
+    submit = SubmitField('Register')
+
+
+class LoginForm(FlaskForm):
+    email = StringField('Email', validators=[DataRequired(), Email()])
+    password = PasswordField('Password', validators=[DataRequired()])
+    # remember_me = BooleanField('Remember me')
+    submit = SubmitField('Login')

connection.py

@@ -0,0 +1,49 @@
+import os
+import psycopg2
+import psycopg2.extras
+
+
+def get_connection_string():
+    # setup connection string
+    # to do this, please define these environment variables first
+    user_name = os.environ.get('PSQL_USER_NAME')
+    password = os.environ.get('PSQL_PASSWORD')
+    host = os.environ.get('PSQL_HOST')
+    database_name = os.environ.get('PSQL_DB_NAME')
+
+    env_variables_defined = user_name and password and host and database_name
+
+    if env_variables_defined:
+        # this string describes all info for psycopg2 to connect to the database
+        return 'postgresql://{user_name}:{password}@{host}/{database_name}'.format(
+            user_name=user_name,
+            password=password,
+            host=host,
+            database_name=database_name
+        )
+    else:
+        raise KeyError('Some necessary environment variable(s) are not defined')
+
+
+def open_database():
+    try:
+        connection_string = get_connection_string()
+        connection = psycopg2.connect(connection_string)
+        connection.autocommit = True
+    except psycopg2.DatabaseError as exception:
+        print('Database connection problem')
+        raise exception
+    return connection
+
+
+def connection_handler(function):
+    def wrapper(*args, **kwargs):
+        connection = open_database()
+        # we set the cursor_factory parameter to return with a RealDictCursor cursor (cursor which provide dictionaries)
+        dict_cur = connection.cursor(cursor_factory=psycopg2.extras.RealDictCursor)
+        ret_value = function(dict_cur, *args, **kwargs)
+        dict_cur.close()
+        connection.close()
+        return ret_value
+
+    return wrapper

data_manager/dm_answers.py

@@ -0,0 +1,91 @@
+from connection import connection_handler
+import datetime
+from data_manager import dm_general
+
+
+@connection_handler
+def get_all_sql_answers_by_question_id(cursor, question_id):
+    cursor.execute("""
+                    SELECT answer.*, users.username FROM answer
+                    LEFT JOIN users ON answer.users_id = users.id
+                    WHERE question_id=%(question_id)s 
+                    ORDER BY is_accepted DESC, vote_number DESC, submission_time ASC
+                    """,
+                    {"question_id": question_id})
+    answers = cursor.fetchall()
+    return answers
+
+
+@connection_handler
+def qet_answer_by_id(cursor, answer_id):
+    cursor.execute("""
+                    SELECT * FROM answer WHERE id=%(answer_id)s
+                    """,
+                    {"answer_id": answer_id})
+    answer = cursor.fetchone()
+    return answer
+
+
+@connection_handler
+def qet_users_id_by_answer_id(cursor, answer_id):
+    cursor.execute("""
+                    SELECT users_id FROM answer WHERE id=%(answer_id)s
+                    """,
+                    {"answer_id": answer_id})
+    answer = cursor.fetchone()
+    return answer['users_id']
+
+
+@connection_handler
+def add_sql_answer(cursor, form_data, question_id, user_id):
+    time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    cursor.execute("""
+                    INSERT INTO answer (users_id, submission_time, vote_number, question_id, is_accepted, message)
+                    VALUES (%(users_id)s, %(time)s, 0, %(question_id)s, 0, %(answer)s)
+                    """, {'time': time, 'question_id': question_id, 'answer': form_data['answer'], 'users_id': user_id})
+
+
+@connection_handler
+def delete_answer(cursor, answer_id):
+    # delete answer image if exists
+    dm_general.remove_image('answer', answer_id)
+
+    # delete answer and all related comments
+    cursor.execute(
+                """
+            DELETE FROM comment WHERE answer_id=%(answer_id)s;
+            DELETE FROM answer WHERE id=%(answer_id)s;
+                """, {'answer_id': answer_id}
+            )
+
+
+@connection_handler
+def update_answer(cursor, answer_id, form_data):
+    cursor.execute(
+        """
+            UPDATE answer SET message=%(answer)s WHERE id=%(answer_id)s
+        """, {"answer": form_data['answer'], "answer_id": answer_id}
+    )
+
+
+@connection_handler
+def change_answer_vote(cursor, answer_id, value_to_change_vote):
+    cursor.execute("""
+                    UPDATE answer
+                    SET vote_number = vote_number + %(value_to_change_vote)s
+                    WHERE id = %(answer_id)s
+                    """, {'value_to_change_vote': value_to_change_vote, 'answer_id': answer_id})
+    cursor.execute("""
+                    SELECT vote_number FROM answer WHERE id=%(answer_id)s
+                    """, {'answer_id': answer_id})
+    new_vote_number = cursor.fetchone()
+    return new_vote_number
+
+
+@connection_handler
+def accept_answer(cursor, answer_id, is_accepted):
+    cursor.execute(
+        """
+            UPDATE answer SET is_accepted=%(is_accepted)s WHERE id=%(answer_id)s
+        """, {"is_accepted": is_accepted, "answer_id": answer_id}
+    )

data_manager/dm_comments.py

@@ -0,0 +1,86 @@
+from connection import connection_handler
+import datetime
+
+
+@connection_handler
+def add_comment_to_question(cursor, form_data, question_id, users_id):
+    time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    cursor.execute("""
+                    INSERT INTO comment
+                    (edited_count, message, question_id, submission_time, users_id)
+                    VALUES (0, %(message)s, %(question_id)s, %(time)s, %(users_id)s)
+                    """,
+                   {'message': form_data['message'], 'question_id': question_id, 'time': time,
+                    'users_id': users_id})
+
+
+@connection_handler
+def add_comment_to_answer(cursor, form_data, answer_id, question_id, users_id):
+    time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    cursor.execute("""
+                    INSERT INTO comment
+                    (edited_count, message, answer_id, submission_time, question_id, users_id)
+                    VALUES (0, %(message)s, %(answer_id)s, %(time)s, %(question_id)s, %(users_id)s)
+                    """,
+                   {'message': form_data['message'], 'answer_id': answer_id,
+                    'time': time, 'question_id': question_id, 'users_id': users_id})
+
+
+@connection_handler
+def show_question_comments_by_id(cursor, question_id):
+    cursor.execute("""
+                    SELECT c.*, u.username FROM comment as c
+                    LEFT JOIN users as u ON c.users_id = u.id
+                    WHERE   question_id = %(question_id)s AND answer_id IS NULL
+                    """,
+                   {'question_id': question_id})
+    comments = cursor.fetchall()
+    return comments
+
+
+@connection_handler
+def show_answer_comments_by_id(cursor, question_id):
+    cursor.execute("""
+                    SELECT c.*, u.username FROM comment as c
+                    LEFT JOIN users as u ON c.users_id = u.id
+                    WHERE question_id = %(question_id)s;  
+                    """,
+                   {'question_id': question_id})
+    comments = cursor.fetchall()
+    return comments
+
+
+@connection_handler
+def get_comment_by_id(cursor, comment_id):
+    cursor.execute("""
+                    SELECT c.*, u.username FROM comment as c
+                    LEFT JOIN users as u ON c.users_id = u.id                    
+                    WHERE c.id = %(comment_id)s;
+                    """,
+                   {'comment_id': comment_id})
+    comment = cursor.fetchone()
+    return comment
+
+
+@connection_handler
+def update_comment_by_id(cursor, form_data, comment_id):
+    time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    cursor.execute("""
+                    UPDATE comment
+                    SET message = %(message)s, edited_count = edited_count + 1, submission_time = %(time)s
+                    WHERE id = %(comment_id)s; 
+                    SELECT question_id FROM comment
+                    WHERE id = %(comment_id)s;
+                    """,
+                   {'message': form_data['message'], 'time': time, 'comment_id': comment_id})
+    question_id = cursor.fetchone()
+    return question_id
+
+
+@connection_handler
+def delete_comment_by_id(cursor, comment_id):
+    cursor.execute("""
+                    DELETE FROM comment
+                    WHERE id = %(comment_id)s;
+                    """,
+                   {'comment_id': comment_id})

data_manager/dm_credentials.py

@@ -0,0 +1,24 @@
+from connection import connection_handler
+
+
+@connection_handler
+def register_user(cursor, form_data, hashed_password):
+    # form_data.username.data, form_data.email.data etc...
+    cursor.execute("""
+                    INSERT INTO users (username, name, email, password)
+                    VALUES (%(username)s, %(name)s, %(email)s, %(password)s)
+                    """, {'username': form_data.username.data,
+                          'name': form_data.name.data,
+                          'email': form_data.email.data,
+                          'password': hashed_password})
+
+
+@connection_handler
+def get_user_data(cursor, email):
+    cursor.execute("""
+                    SELECT * FROM users
+                    WHERE email = %(email)s
+                    """,
+                   {'email': email})
+    user_data = cursor.fetchone()
+    return user_data

data_manager/dm_general.py

@@ -0,0 +1,38 @@
+from connection import connection_handler
+import os
+from psycopg2 import sql
+
+
+@connection_handler
+def update_image_path(cursor, table, filename, id):
+    cursor.execute(f"""
+                    UPDATE {table} 
+                    SET image = 'images/{filename}' WHERE id={id}
+                    """)
+
+
+@connection_handler
+def remove_image(cursor, table_name, resource_id):
+    cursor.execute(
+        sql.SQL('select image from {} WHERE id=%s').format(sql.Identifier(table_name)), [resource_id]
+    )
+    file_path = cursor.fetchone()
+    if file_path and file_path['image'] != None:
+        os.remove(f"./static/{file_path['image']}")
+
+
+@connection_handler
+def search_results(cursor, search_term):
+
+    cursor.execute("""
+            SELECT DISTINCT q.* FROM question q JOIN answer a ON q.id=a.question_id WHERE
+            UPPER(q.title) LIKE UPPER(%(search_term)s)
+            OR UPPER(q.message) LIKE UPPER(%(search_term)s)
+            OR UPPER(a.message) LIKE UPPER(%(search_term)s);
+                    """, {'search_term': '%'+search_term+'%'})
+    questions = cursor.fetchall()
+    cursor.execute("""
+            SELECT * FROM answer WHERE UPPER(message) LIKE UPPER(%(search_term)s)
+                    """, {'search_term': '%'+search_term+'%'})
+    answers = cursor.fetchall()
+    return questions, answers