|
import os |
|
from line_data_retriever import LineDataRetriever |
|
from image_saver import ImageSaver |
|
from s3_uploader import S3Uploader |
|
|
|
def main(): |
|
line_api_token = os.environ["LINE_API_TOKEN"] |
|
aws_access_key_id = os.environ["AWS_ACCESS_KEY_ID"] |
|
aws_secret_access_key = os.environ["AWS_SECRET_ACCESS_KEY"] |
|
bucket_name = os.environ["S3_BUCKET_NAME"] |
|
|
|
retriever = LineDataRetriever(line_api_token) |
|
data = retriever.retrieve_data() |
|
|
|
for message in data["messages"]: |
|
if message["type"] == "image": |
|
image_data = message["image"] |
|
saver = ImageSaver(image_data) |
|
file_path = "temp.jpg" |
|
saver.save_image(file_path) |
|
|
|
uploader = S3Uploader(aws_access_key_id, aws_secret_access_key, bucket_name) |
|
key = f"images/{os.path.basename(file_path)}" |
|
uploader.upload_file(file_path, key) |
|
|
|
os.remove(file_path) |
|
|
|
if __name__ == "__main__": |
|
main() |