-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.py
More file actions
38 lines (31 loc) · 1.27 KB
/
main.py
File metadata and controls
38 lines (31 loc) · 1.27 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import json
import csv
import boto3
import colorsys
def handler_color(event, context):
# Pobieramy listę plików csv z folderu csv-files w buckecie csv-bucket
s3 = boto3.client('s3')
bucket_name = 'csv-bucket'
prefix = 'csv-files/'
file_names = ['example.csv', 'example1.csv', 'example2.csv']
# Pobieramy zawartość plików csv i tworzymy listę słowników z danymi o kolorach
color_data = []
for file_name in file_names:
csv_content = s3.get_object(Bucket=bucket_name, Key=f'{prefix}{file_name}')['Body'].read().decode()
reader = csv.DictReader(csv_content.splitlines())
for row in reader:
color_hex = row['value']
r, g, b = colorsys.hex_to_rgb(color_hex)
color_data.append({
'name': row['color'],
'hex': color_hex,
'rgb': [r, g, b]
})
# Tworzymy plik json z danymi o kolorach i zapisujemy go do bucket'u
json_data = json.dumps(color_data)
s3.put_object(Bucket=bucket_name, Key='json-files/colors.json', Body=json_data)
# Wyświetlamy dostępną paletę kolorów
print('Dostępna paleta kolorów:')
for color in color_data:
print(f'{color["name"]}: #{color["hex"]} ({color["rgb"]})')
return len(color_data)