Автоматическая рассылка файлов по электронной почте

import os
import base64
import pickle
import time
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from email.mime.base import MIMEBase
from email import encoders
from google.auth.transport.requests import Request
from google_auth_oauthlib.flow import InstalledAppFlow
from googleapiclient.discovery import build
from googleapiclient.errors import HttpError
 
 
# Define the scope of access
SCOPES = ['https://www.googleapis.com/auth/gmail.send']
 
# Function to authenticate and get service object
def authenticate_gmail():
    """Authenticate to Gmail API and return service object."""
    creds = None
    # Check if token.json exists for previously saved credentials
    if os.path.exists('token.json'):
        with open('token.json', 'rb') as token:
            creds = pickle.load(token)
    
    # If no valid credentials are available, let the user log in
    if not creds or not creds.valid:
        if creds and creds.expired and creds.refresh_token:
            creds.refresh(Request())
        else:
            flow = InstalledAppFlow.from_client_secrets_file(
                'credentials.json', SCOPES)
            creds = flow.run_local_server(port=8080)
        
        # Save the credentials for future use
        with open('token.json', 'wb') as token:
            pickle.dump(creds, token)
 
    try:
        service = build('gmail', 'v1', credentials=creds)
        return service
    except HttpError as error:
        print(f'An error occurred: {error}')
        return None
 
# Function to create a message with attachments
def create_message_with_attachment(sender, to, subject, message_text, file_path):
    """Create a MIME message with attachment."""
    message = MIMEMultipart()
    message['to'] = to
    message['from'] = sender
    message['subject'] = subject
    
    # Add the text part of the message
    msg = MIMEText(message_text)
    message.attach(msg)
    
    # Attach the file
    try:
        filename = os.path.basename(file_path)
        with open(file_path, 'rb') as f:
            file_data = f.read()
            file_type = 'application/octet-stream'
            part = MIMEBase('application', 'octet-stream')
            part.set_payload(file_data)
            encoders.encode_base64(part)
            part.add_header('Content-Disposition', f'attachment; filename={filename}')
            message.attach(part)
    except Exception as e:
        print(f"Error attaching file {file_path}: {str(e)}")
    
    raw_message = base64.urlsafe_b64encode(message.as_bytes()).decode()
    return {'raw': raw_message}
 
# Function to send email
def send_email(service, sender, to, subject, message_text, file_path):
    """Send an email using Gmail API."""
    try:
        message = create_message_with_attachment(sender, to, subject, message_text, file_path)
        send_message = service.users().messages().send(userId="me", body=message).execute()
        print(f"Message sent to {to}")
        return send_message
    except HttpError as error:
        print(f'An error occurred: {error}')
 
# Function to send bulk emails to predefined list of recipients
def send_bulk_emails(service, sender, recipients, subject, message_text, file_paths):
    """Send bulk emails with attachments."""
    for recipient in recipients:
        for file_path in file_paths:
            send_email(service, sender, recipient, subject, message_text, file_path)
            time.sleep(1)  # Sleep to avoid Gmail rate limits
 
# Main function
def main():
    sender = "your-email@gmail.com"
    subject = "Subject of your email"
    message_text = "This is the body of the email."
    
    # List of predefined recipients
    recipients = [
        "client1@example.com",
        "client2@example.com",
        "client3@example.com",
    ]
    
    # List of files you want to send
    file_paths = [
        "path/to/your/file.xlsx",
        "path/to/your/file.docx",
        "path/to/your/file.pdf"
    ]
    
    # Authenticate and get service object
    service = authenticate_gmail()
    if service:
        send_bulk_emails(service, sender, recipients, subject, message_text, file_paths)
 
if __name__ == '__main__':
    main()