
    Vhd                     R    S SK Jr  S SKrS SKJrJr  S rS rS r	S r
S rS	 rS
 rg)    )sessionN)check_password_hashgenerate_password_hashc            	      h    [         R                  " SSSSS[         R                  R                  S9n U $ )N	localhostfigoscolaridb_figoutf8mb4)hostuserpassworddatabasecharsetcursorclass)pymysqlconnectcursors
DictCursor)dbs    -/Users/jbastos/public_html/wsgi/Uploadr/db.pyget_dbr      s3    	OO..
0B I    c                     S n [        5       nUR                  5       R                  SU [        U5      45        UR	                  5         U$ ! WR
                   a
    SU  S3n U$ f = f)Nz6INSERT INTO users (username, password) VALUES (%s, %s)zUser z is already registered.)r   cursorexecuter   commitIntegrityError)usernamer   errorr   s       r   register_userr!      su    E:X
		D-h78	
 			 L  :z!89L:s   AA A%$A%c                     S n[        5       R                  5       nUR                  SU 45        UR                  5       nUc  SnU$ [	        US   U5      (       d  SnU$ [
        R                  " 5         US   [
        S'   U$ )Nz'SELECT * FROM users WHERE username = %szIncorrect credentials.r   iduser_id)r   r   r   fetchoner   r   clear)r   r   r    r   r   s        r   
login_userr'      s    E		BJJ1H; ;;=D|( L !j!18<<( L 	!$Z	Lr   c                     [        5       R                  5       nSU  S3nUR                  U5        UR                  5       $ )Nz!SELECT * FROM files WHERE owner='')r   r   r   fetchall)r$   r   querys      r   	get_filesr,   3   s6    		B/y:EJJu;;=r   c                     [        5       R                  5       nSU  S3nUR                  U5        UR                  5       nUS   $ )NzSELECT * FROM files WHERE id='r)   name)r   r   r   r%   )r#   r   r+   files       r   get_file_namer0   9   s?    		B,RD2EJJu;;=D<r   c                     S n [        5       nUR                  5       R                  SU [        S   45        UR	                  5         U$ ! WR
                   a
    SU  S3n U$ f = f)Nz/INSERT INTO files (name, owner) VALUES (%s, %s)r$   zFile 'z' is already uploaded.)r   r   r   r   r   r   )r.   r    r   s      r   file_insertr2   @   st    E6X
		=79%&	
 			 L  645L6s   AA	 	A#"A#c                 x    [        5       nUR                  5       R                  SU 5        UR                  5         g )NzDELETE FROM files WHERE id = %s)r   r   r   r   )r#   r   s     r   file_remover4   M   s.    	BIIK)	 IIKr   )flaskr   pymysql.cursorsr   werkzeug.securityr   r   r   r!   r'   r,   r0   r2   r4    r   r   <module>r9      s1      I(r   