forked from scribe-org/Scribe-Data
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request scribe-org#480 from axif0/pyICU
(New) Fixed pyICU capabilities for emoji functionalities
- Loading branch information
Showing
3 changed files
with
264 additions
and
34 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,211 @@ | ||
""" | ||
Check to see if the requirements of the emoji process are installed. | ||
.. raw:: html | ||
<!-- | ||
* Copyright (C) 2024 Scribe | ||
* | ||
* This program is free software: you can redistribute it and/or modify | ||
* it under the terms of the GNU General Public License as published by | ||
* the Free Software Foundation, either version 3 of the License, or | ||
* (at your option) any later version. | ||
* | ||
* This program is distributed in the hope that it will be useful, | ||
* but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
* GNU General Public License for more details. | ||
* | ||
* You should have received a copy of the GNU General Public License | ||
* along with this program. If not, see <https://www.gnu.org/licenses/>. | ||
--> | ||
""" | ||
|
||
import os | ||
import platform # added to check the OS | ||
import subprocess | ||
import sys | ||
from pathlib import Path | ||
|
||
import pkg_resources | ||
import requests | ||
|
||
|
||
def check_if_pyicu_installed(): | ||
installed_packages = {pkg.key for pkg in pkg_resources.working_set} | ||
|
||
return "pyicu" in installed_packages | ||
|
||
|
||
def get_python_version_and_architecture(): | ||
""" | ||
Get the current Python version and architecture. | ||
Returns | ||
------- | ||
python_version : str | ||
The Python version in the format 'cpXY'. | ||
architecture : str | ||
The architecture type ('amd64' or 'win32'). | ||
""" | ||
version = sys.version_info | ||
python_version = f"cp{version.major}{version.minor}" | ||
architecture = "win_amd64" if sys.maxsize > 2**32 else "win32" | ||
return python_version, architecture | ||
|
||
|
||
def fetch_wheel_releases(): | ||
""" | ||
Fetch the release data for PyICU from GitHub. | ||
Returns | ||
------- | ||
available_wheels : list | ||
A list of tuples containing wheel file names and their download URLs. | ||
total_size_mb : float | ||
The total size of all available wheels in MB. | ||
""" | ||
url = "https://api.github.com/repos/cgohlke/pyicu-build/releases" | ||
response = requests.get(url) | ||
response.raise_for_status() # raise an error for bad responses | ||
|
||
available_wheels = [] | ||
total_size_bytes = 0 | ||
|
||
for release in response.json(): | ||
for asset in release["assets"]: | ||
if asset["name"].endswith(".whl"): | ||
available_wheels.append((asset["name"], asset["browser_download_url"])) | ||
total_size_bytes += asset["size"] | ||
|
||
total_size_mb = total_size_bytes / (1024 * 1024) # convert bytes to MB | ||
return available_wheels, total_size_mb | ||
|
||
|
||
def download_wheel_file(wheel_url, output_dir): | ||
""" | ||
Download the wheel file from the given URL. | ||
Parameters | ||
---------- | ||
wheel_url : str | ||
The URL of the wheel file to download. | ||
output_dir : str | ||
The directory to save the downloaded file. | ||
Returns | ||
------- | ||
str : path to the downloaded wheel file. | ||
""" | ||
response = requests.get(wheel_url) | ||
response.raise_for_status() # raise an error for bad responses | ||
|
||
wheel_filename = os.path.basename(wheel_url) | ||
wheel_path = os.path.join(output_dir, wheel_filename) | ||
|
||
with open(wheel_path, "wb") as wheel_file: | ||
wheel_file.write(response.content) | ||
|
||
return wheel_path | ||
|
||
|
||
def find_matching_wheel(wheels, python_version, architecture): | ||
""" | ||
Find the matching wheel file based on Python version and architecture. | ||
Parameters | ||
---------- | ||
wheels : list | ||
The list of available wheels. | ||
python_version : str | ||
The Python version (e.g., 'cp311'). | ||
architecture : str | ||
The architecture type (e.g., 'win_amd64'). | ||
Returns | ||
------- | ||
str : The download URL of the matching wheel or None if not found. | ||
""" | ||
return next( | ||
( | ||
download_url | ||
for name, download_url in wheels | ||
if python_version in name and architecture in name | ||
), | ||
None, | ||
) | ||
|
||
|
||
def check_and_install_pyicu(): | ||
package_name = "PyICU" | ||
installed_packages = {pkg.key for pkg in pkg_resources.working_set} | ||
if package_name.lower() not in installed_packages: | ||
# print(f"{package_name} not found. Installing...") | ||
|
||
# Fetch available wheels from GitHub to estimate download size. | ||
wheels, total_size_mb = fetch_wheel_releases() | ||
|
||
print( | ||
f"{package_name} is not installed.\nIt will be downloaded from 'https://github.com/repos/cgohlke/pyicu'" | ||
f"\nApproximately {total_size_mb:.2f} MB will be downloaded.\nDo you want to proceed? (Y/n)?" | ||
) | ||
|
||
user_input = input().strip().lower() | ||
if user_input in ["", "y", "yes"]: | ||
print("Proceeding with installation...") | ||
|
||
else: | ||
print("Installation aborted by the user.") | ||
return False | ||
|
||
# Check the operating system. | ||
if platform.system() != "Windows": | ||
# If not Windows, directly use pip to install PyICU. | ||
try: | ||
subprocess.run( | ||
[sys.executable, "-m", "pip", "install", package_name], check=True | ||
) | ||
print(f"{package_name} has been installed successfully.") | ||
|
||
except subprocess.CalledProcessError as e: | ||
print(f"Error occurred while installing {package_name}: {e}") | ||
return False | ||
|
||
else: | ||
# Windows-specific installation using wheel files. | ||
python_version, architecture = get_python_version_and_architecture() | ||
|
||
# Find the matching wheel for the current Python version and architecture. | ||
wheel_url = find_matching_wheel(wheels, python_version, architecture) | ||
|
||
if not wheel_url: | ||
print( | ||
"No matching wheel file found for your Python version and architecture." | ||
) | ||
return False | ||
|
||
# Download the wheel file. | ||
output_dir = Path.cwd() # use the current directory for simplicity | ||
wheel_path = download_wheel_file(wheel_url, output_dir) | ||
|
||
# Install PyICU using pip. | ||
try: | ||
subprocess.run( | ||
[sys.executable, "-m", "pip", "install", wheel_path], | ||
check=True, | ||
) | ||
print(f"{package_name} has been installed successfully.") | ||
|
||
# Remove the downloaded wheel file. | ||
os.remove(wheel_path) | ||
print(f"Removed temporary file: {wheel_path}") | ||
|
||
except subprocess.CalledProcessError as e: | ||
print(f"Error occurred while installing {package_name}: {e}") | ||
return False | ||
|
||
return True |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters