yeref 0.24.97__py3-none-any.whl → 0.24.99__py3-none-any.whl

yeref/yeref.py

@@ -16011,100 +16011,108 @@ async def calc_metrics(bot, PROJECT_USERNAME, dataroom_folder_id, EXTRA_D, BASE_
         r4 = await return_retention_metrics(bot, data_users, EXTRA_D)
         r5 = await return_profit_and_loss_metrics(bot, data_users, EXTRA_D)
 
-        if r1 and os.path.isfile(r1):
-            # 1) Авторизация в Google Sheets
-            scopes = r_conf('scopes', CONF_P)
-            credential_path = os.path.join(EXTRA_D, (r_conf('credential_file', CONF_P))[0])
-            credentials = ServiceAccountCredentials.from_json_keyfile_name(credential_path, scopes)
-            http_auth = credentials.authorize(httplib2.Http())
-            sheets_service = build('sheets', 'v4', http=http_auth, cache_discovery=False)
-
-            # 2) Получаем metadata всей таблицы, чтобы увидеть, какие листы есть сейчас
-            spreadsheet = sheets_service.spreadsheets().get(spreadsheetId=dataroom_folder_id).execute()
-            logger.info(f"До переименования. Найдено листов: {len(spreadsheet.get('sheets', []))}")
-            for sheet in spreadsheet.get('sheets', []):
-                props = sheet.get('properties', {})
-                logger.info(f"  • sheetId={props.get('sheetId')}  title='{props.get('title')}'")
-
-            # 3) Ищем лист с названием «Лист1» или «Sheet1»
-            sheet_id_to_rename = None
-            original_title = None
-            for sheet in spreadsheet.get('sheets', []):
-                props = sheet.get('properties', {})
-                title = props.get('title', '')
-                if title in ['Лист1', 'Sheet1']:
-                    sheet_id_to_rename = props.get('sheetId')
-                    original_title = title
-                    logger.info(f"Найден лист для переименования: sheetId={sheet_id_to_rename}, title='{title}'")
-                    break
+    # --- Начинаем сразу с объединённого цикла по всем метрикам r1–r5 ---
+    metrics_paths = [r1, r2, r3, r4, r5]
+    tables = []  # сюда будем складывать словари вида {'name': basename, 'rows': [[...], ...]}
+
+    # 1) Собираем существующие CSV-файлы
+    for path in metrics_paths:
+        if path and os.path.isfile(path):
+            basename = os.path.basename(path)
+            with open(path, newline='', encoding='utf-8') as csvfile:
+                reader = csv.reader(csvfile)
+                rows = list(reader)
+            logger.info(f"Найден файл '{basename}', строк = {len(rows)}.")
+            tables.append({'name': basename, 'rows': rows})
+        else:
+            logger.warning(f"Файл не найден или не существует: {path}")
 
-            # 4) Если нашли, то переименовываем в PROJECT_USERNAME
-            if sheet_id_to_rename is not None:
-                rename_request = {
-                    'requests': [
-                        {
-                            'updateSheetProperties': {
-                                'properties': {
-                                    'sheetId': sheet_id_to_rename,
-                                    'title': PROJECT_USERNAME
-                                },
-                                'fields': 'title'
-                            }
-                        }
-                    ]
-                }
-                sheets_service.spreadsheets().batchUpdate(
-                    spreadsheetId=dataroom_folder_id,
-                    body=rename_request
-                ).execute()
-                logger.info(f"Запрос на переименование отправлен: '{original_title}' → '{PROJECT_USERNAME}'")
-
-                # 5) Подтягиваем снова metadata, чтобы убедиться, что название поменялось
-                spreadsheet_after = sheets_service.spreadsheets().get(spreadsheetId=dataroom_folder_id).execute()
-                logger.info("После переименования. Список листов теперь:")
-                for sheet in spreadsheet_after.get('sheets', []):
-                    props = sheet.get('properties', {})
-                    logger.info(f"  • sheetId={props.get('sheetId')}  title='{props.get('title')}'")
-            else:
-                logger.warning("Не найден лист с названием 'Лист1' или 'Sheet1' — пропускаем переименование.")
+    # 2) Если ни одного CSV не обнаружено, выходим
+    if not tables:
+        logger.warning("Ни один CSV-файл не найден. Прерываем запись.")
+    else:
+        # 3) Авторизация в Google Sheets
+        scopes = r_conf('scopes', CONF_P)
+        credential_path = os.path.join(EXTRA_D, (r_conf('credential_file', CONF_P))[0])
+        credentials = ServiceAccountCredentials.from_json_keyfile_name(credential_path, scopes)
+        http_auth = credentials.authorize(httplib2.Http())
+        sheets_service = build('sheets', 'v4', http=http_auth, cache_discovery=False)
 
-            # 6) Читаем CSV-файл (r1) целиком
-            with open(r1, newline='', encoding='utf-8') as csvfile:
-                reader = csv.reader(csvfile)
-                all_rows = list(reader)
-            logger.info(f"Прочитано {len(all_rows)} строк из файла {r1}.")
-
-            # 7) Составляем диапазон для записи: используем PROJECT_USERNAME
-            #    Если название листа может содержать пробелы или спецсимволы, оборачиваем в одинарные кавычки:
-            safe_title = PROJECT_USERNAME.replace("'", "\\'")
-            range_str = f"'{safe_title}'!A1"
-            logger.info(f"Будем записывать в диапазон: {range_str}")
-
-            # 8) Формируем тело запроса на запись
-            write_body = {
-                'valueInputOption': 'USER_ENTERED',
-                'data': [
+        # 4) Переименовываем стандартный лист «Лист1»/«Sheet1» → PROJECT_USERNAME
+        spreadsheet = sheets_service.spreadsheets().get(spreadsheetId=dataroom_folder_id).execute()
+        sheet_id_to_rename = None
+        for sheet in spreadsheet.get('sheets', []):
+            props = sheet.get('properties', {})
+            title = props.get('title', '')
+            if title in ['Лист1', 'Sheet1']:
+                sheet_id_to_rename = props.get('sheetId')
+                break
+
+        if sheet_id_to_rename is not None:
+            rename_request = {
+                'requests': [
                     {
-                        'range': range_str,
-                        'majorDimension': 'ROWS',
-                        'values': all_rows
+                        'updateSheetProperties': {
+                            'properties': {
+                                'sheetId': sheet_id_to_rename,
+                                'title': PROJECT_USERNAME
+                            },
+                            'fields': 'title'
+                        }
                     }
                 ]
             }
-
-            # 9) Делаем batchUpdate, чтобы записать всё из all_rows
-            try:
-                response = sheets_service.spreadsheets().values().batchUpdate(
-                    spreadsheetId=dataroom_folder_id,
-                    body=write_body
-                ).execute()
-                updated_cells = response.get('totalUpdatedCells', 0)
-                logger.info(f"Успешно записано в Google Sheets: обновлено ячеек = {updated_cells}.")
-            except Exception as e:
-                logger.error(f"Ошибка при записи в Google Sheets: {e}")
+            sheets_service.spreadsheets().batchUpdate(
+                spreadsheetId=dataroom_folder_id,
+                body=rename_request
+            ).execute()
+            logger.info(f"Лист переименован в '{PROJECT_USERNAME}'.")
         else:
-            logger.warning(f"Файл с метриками не найден или не существует: {r1}")
+            logger.warning("Не найден лист 'Лист1' или 'Sheet1' для переименования.")
+
+        # 5) Готовим запросы для записи всех таблиц одна за другой с промежутком
+        safe_title = PROJECT_USERNAME.replace("'", "\\'")
+        offset = 1  # первая строка, с которой начнём писать
+        data_requests = []
+
+        for tbl in tables:
+            # 5.1) Заголовок: имя CSV-файла
+            header_range = f"'{safe_title}'!A{offset}"
+            data_requests.append({
+                'range': header_range,
+                'majorDimension': 'ROWS',
+                'values': [[tbl['name']]]
+            })
+            logger.info(f"Добавлен заголовок '{tbl['name']}' → A{offset}")
+            offset += 1
+
+            # 5.2) Содержимое CSV: все строки сразу под заголовком
+            data_range = f"'{safe_title}'!A{offset}"
+            data_requests.append({
+                'range': data_range,
+                'majorDimension': 'ROWS',
+                'values': tbl['rows']
+            })
+            logger.info(f"Добавлены {len(tbl['rows'])} строк из '{tbl['name']}' → A{offset}")
+            offset += len(tbl['rows'])
+
+            # 5.3) Оставляем минимум 2 пустые строки перед следующей таблицей
+            offset += 2
 
+        # 6) Выполняем единый batchUpdate для всех data_requests
+        write_body = {
+            'valueInputOption': 'USER_ENTERED',
+            'data': data_requests
+        }
+        try:
+            response = sheets_service.spreadsheets().values().batchUpdate(
+                spreadsheetId=dataroom_folder_id,
+                body=write_body
+            ).execute()
+            total_cells = response.get('totalUpdatedCells', 0)
+            logger.info(f"Успешно записано в Google Sheets: всего обновлено ячеек = {total_cells}.")
+        except Exception as e:
+            logger.error(f"Ошибка при записи всех таблиц в Google Sheets: {e}")
     except Exception as e:
         logger.info(log_ % str(e))
         await asyncio.sleep(round(random.uniform(0, 1), 2))

{yeref-0.24.97.dist-info → yeref-0.24.99.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: yeref
-Version: 0.24.97
+Version: 0.24.99
 Summary: desc-f
 Author: john smith
 Dynamic: author

yeref-0.24.99.dist-info/RECORD

@@ -0,0 +1,8 @@
+yeref/__init__.py,sha256=Qpv3o6Xa78VdLcsSRmctGtpnYE9btpAkCekgGhgJyXM,49
+yeref/l_.py,sha256=LMX_olmJwq-tgoALJCnhV_fGrL_i_43yBLkLIcEVqGo,1176743
+yeref/tonweb.js,sha256=Jf6aFOQ1OIY4q7fINYz-m5LsI3seMus124M5SYYZmtE,443659
+yeref/yeref.py,sha256=Dj2jndTfiXUJ_fB-hL4tP9ougOcVbUxXdE6HHZ-ikEw,1055246
+yeref-0.24.99.dist-info/METADATA,sha256=DEUgBkgS-tNtezhSF4aD_Rh59oyMn13Mbtql4JbRvvQ,119
+yeref-0.24.99.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+yeref-0.24.99.dist-info/top_level.txt,sha256=yCQKchWHbfV-3OuQPYRdi2loypD-nmbDJbtt3OuKKkY,6
+yeref-0.24.99.dist-info/RECORD,,

yeref-0.24.97.dist-info/RECORD

@@ -1,8 +0,0 @@
-yeref/__init__.py,sha256=Qpv3o6Xa78VdLcsSRmctGtpnYE9btpAkCekgGhgJyXM,49
-yeref/l_.py,sha256=LMX_olmJwq-tgoALJCnhV_fGrL_i_43yBLkLIcEVqGo,1176743
-yeref/tonweb.js,sha256=Jf6aFOQ1OIY4q7fINYz-m5LsI3seMus124M5SYYZmtE,443659
-yeref/yeref.py,sha256=jdG1SO9WRV8e0o8HjviTOMG-m4rxya1lcClakmVDVZc,1055625
-yeref-0.24.97.dist-info/METADATA,sha256=MNCnHGJUW1SZaDvGSRhQClVxHOkxpX0KrrOI50o-aDM,119
-yeref-0.24.97.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-yeref-0.24.97.dist-info/top_level.txt,sha256=yCQKchWHbfV-3OuQPYRdi2loypD-nmbDJbtt3OuKKkY,6
-yeref-0.24.97.dist-info/RECORD,,