@ -45,6 +45,9 @@ _logger = logging.getLogger(__name__)
ONEDRIVE_SCOPE = [ ' offline_access openid Files.ReadWrite.All ' ]
MICROSOFT_GRAPH_END_POINT = " https://graph.microsoft.com "
GOOGLE_AUTH_ENDPOINT = ' https://accounts.google.com/o/oauth2/auth '
GOOGLE_TOKEN_ENDPOINT = ' https://accounts.google.com/o/oauth2/token '
GOOGLE_API_BASE_URL = ' https://www.googleapis.com '
class AutoDatabaseBackup ( models . Model ) :
@ -100,6 +103,118 @@ class AutoDatabaseBackup(models.Model):
onedrive_folder_id = fields . Char ( string = ' Folder ID ' )
is_onedrive_token_generated = fields . Boolean ( string = ' onedrive Tokens Generated ' ,
compute = ' _compute_is_onedrive_token_generated ' , copy = False )
gdrive_refresh_token = fields . Char ( string = ' Google drive Refresh Token ' , copy = False )
gdrive_access_token = fields . Char ( string = ' Google Drive Access Token ' , copy = False )
is_google_drive_token_generated = fields . Boolean ( string = ' Google drive Token Generated ' ,
compute = ' _compute_is_google_drive_token_generated ' , copy = False )
gdrive_client_id = fields . Char ( string = ' Google Drive Client ID ' , copy = False )
gdrive_client_secret = fields . Char ( string = ' Google Drive Client Secret ' , copy = False )
gdrive_token_validity = fields . Datetime ( string = ' Google Drive Token Validity ' , copy = False )
gdrive_redirect_uri = fields . Char ( string = ' Google Drive Redirect URI ' , compute = ' _compute_redirect_uri ' )
onedrive_redirect_uri = fields . Char ( string = ' Onedrive Redirect URI ' , compute = ' _compute_redirect_uri ' )
@api . depends ( ' backup_destination ' )
def _compute_redirect_uri ( self ) :
for rec in self :
base_url = request . env [ ' ir.config_parameter ' ] . get_param ( ' web.base.url ' )
rec . onedrive_redirect_uri = base_url + ' /onedrive/authentication '
rec . gdrive_redirect_uri = base_url + ' /google_drive/authentication '
@api . depends ( ' gdrive_access_token ' , ' gdrive_refresh_token ' )
def _compute_is_google_drive_token_generated ( self ) :
"""
Set True if the Google Drive refresh token is generated
"""
for rec in self :
rec . is_google_drive_token_generated = bool ( rec . gdrive_access_token ) and bool ( rec . gdrive_refresh_token )
def action_get_gdrive_auth_code ( self ) :
"""
Generate ogoogle drive authorization code
"""
action = self . env [ " ir.actions.act_window " ] . sudo ( ) . _for_xml_id ( " auto_database_backup.action_db_backup_configure " )
base_url = request . env [ ' ir.config_parameter ' ] . get_param ( ' web.base.url ' )
url_return = base_url + ' /web#id= %d &action= %d &view_type=form&model= %s ' % ( self . id , action [ ' id ' ] , ' db.backup.configure ' )
state = {
' backup_config_id ' : self . id ,
' url_return ' : url_return
}
encoded_params = urls . url_encode ( {
' response_type ' : ' code ' ,
' client_id ' : self . gdrive_client_id ,
' scope ' : ' https://www.googleapis.com/auth/drive https://www.googleapis.com/auth/drive.file ' ,
' redirect_uri ' : base_url + ' /google_drive/authentication ' ,
' access_type ' : ' offline ' ,
' state ' : json . dumps ( state ) ,
' approval_prompt ' : ' force ' ,
} )
auth_url = " %s ? %s " % ( GOOGLE_AUTH_ENDPOINT , encoded_params )
return {
' type ' : ' ir.actions.act_url ' ,
' target ' : ' self ' ,
' url ' : auth_url ,
}
def generate_gdrive_refresh_token ( self ) :
"""
generate google drive access token from refresh token if expired
"""
headers = { " content-type " : " application/x-www-form-urlencoded " }
data = {
' refresh_token ' : self . gdrive_refresh_token ,
' client_id ' : self . gdrive_client_id ,
' client_secret ' : self . gdrive_client_secret ,
' grant_type ' : ' refresh_token ' ,
}
try :
res = requests . post ( GOOGLE_TOKEN_ENDPOINT , data = data , headers = headers )
res . raise_for_status ( )
response = res . content and res . json ( ) or { }
if response :
expires_in = response . get ( ' expires_in ' )
self . write ( {
' gdrive_access_token ' : response . get ( ' access_token ' ) ,
' gdrive_token_validity ' : fields . Datetime . now ( ) + timedelta ( seconds = expires_in ) if expires_in else False ,
} )
except requests . HTTPError as error :
error_key = error . response . json ( ) . get ( " error " , " nc " )
error_msg = _ (
" An error occurred while generating the token. Your authorization code may be invalid or has already expired [ %s ]. "
" You should check your Client ID and secret on the Google APIs plateform or try to stop and restart your calendar synchronisation. " ,
error_key )
raise UserError ( error_msg )
def get_gdrive_tokens ( self , authorize_code ) :
"""
Generate onedrive tokens from authorization code
"""
base_url = request . env [ ' ir.config_parameter ' ] . get_param ( ' web.base.url ' )
headers = { " content-type " : " application/x-www-form-urlencoded " }
data = {
' code ' : authorize_code ,
' client_id ' : self . gdrive_client_id ,
' client_secret ' : self . gdrive_client_secret ,
' grant_type ' : ' authorization_code ' ,
' redirect_uri ' : base_url + ' /google_drive/authentication '
}
try :
res = requests . post ( GOOGLE_TOKEN_ENDPOINT , params = data ,
headers = headers )
res . raise_for_status ( )
response = res . content and res . json ( ) or { }
if response :
expires_in = response . get ( ' expires_in ' )
self . write ( {
' gdrive_access_token ' : response . get ( ' access_token ' ) ,
' gdrive_refresh_token ' : response . get ( ' refresh_token ' ) ,
' gdrive_token_validity ' : fields . Datetime . now ( ) + timedelta (
seconds = expires_in ) if expires_in else False ,
} )
except requests . HTTPError :
error_msg = _ ( " Something went wrong during your token generation. Maybe your Authorization Code is invalid " )
raise UserError ( error_msg )
@api . depends ( ' onedrive_access_token ' , ' onedrive_refresh_token ' )
def _compute_is_onedrive_token_generated ( self ) :
@ -382,12 +497,14 @@ class AutoDatabaseBackup(models.Model):
client . close ( )
# Google Drive backup
elif rec . backup_destination == ' google_drive ' :
if rec . gdrive_token_validity < = fields . Datetime . now ( ) :
rec . generate_gdrive_refresh_token ( )
temp = tempfile . NamedTemporaryFile ( suffix = ' . %s ' % rec . backup_format )
with open ( temp . name , " wb+ " ) as tmp :
odoo . service . db . dump_db ( rec . db_name , tmp , rec . backup_format )
try :
access_token = self . env [ ' google.drive.config ' ] . sudo ( ) . get_access_token ( )
headers = { " Authorization " : " Bearer %s " % access_token }
# access_token = self.env['google.drive.config'].sudo().get_access_token( )
headers = { " Authorization " : " Bearer %s " % rec . gdrive_ access_token}
para = {
" name " : backup_filename ,
" parents " : [ rec . google_drive_folderid ] ,
AjmalCybro
2 years ago
