check_legal.py

import os
import sys
import base64
import subprocess
from pathlib import Path


class CheckLegalUse:

    def __init__(self,
                filename,
                gist_url,
                dir_name='punch',
                n=1,
                ):
        import FreeCAD
        freecad_dir = Path(FreeCAD.ConfigGet("UserAppData"))
        if not freecad_dir.exists():
            freecad_dir.mkdir()
        application_dir = freecad_dir / dir_name
        if not application_dir.exists():
            application_dir.mkdir()
        self.filename = application_dir / filename
        self.gist_url = gist_url
        self.n = n

    def allowed_to_continue(self):
        if sys.platform == "win32":
            if not self.is_registered:
                try:
                    self.serial = str(subprocess.check_output("wmic csproduct get uuid")).split("\\r\\r\\n")[1].split()[0]
                except Exception as e:
                    print(f"Error retrieving serial number: {e}")
                    try:
                        import wmi
                    except ImportError:
                        from freecad_funcs import install_packages
                        install_packages(['wmi', 'pywin32', 'pypiwin32'])
                        try:
                            import wmi
                        except ImportError:
                            return False, 'REBOOT'
                    # Fallback to wmi package if wmic fails
                    c = wmi.WMI()
                    self.serial = c.Win32_ComputerSystemProduct()[0].UUID
                if not internet():
                    return False, 'INTERNET'
                elif not self.serial_number(self.serial):
                    return False, 'SERIAL'
                else:
                    self.register()
                    return True, 'REGISTERED'
            return True, ''
        return True, ''

    def initiate(self):
        if not self.filename.exists():
            with open(self.filename, 'wb') as f:
                b = base64.b64encode('0-0'.encode('utf-8'))
                f.write(b)

    @property
    def is_registered(self):
        if not Path(self.filename).exists():
            self.initiate()
            return True
        else:
            text = self.get_registered_numbers()
            if text[0] == 1 or text[1] <= self.n:
                return True
            else:
                return False

    def get_registered_numbers(self):
        if not Path(self.filename).exists():
            self.initiate()
        with open(self.filename, 'rb') as f:
            b = f.read()
            text = base64.b64decode(b).decode('utf-8').split('-')
            return int(text[0]), int(text[1])

    def add_using_feature(self):
        if  not Path(self.filename).exists():
            self.initiate()
        with open(self.filename, 'rb') as f:
            b = f.read()
            text = base64.b64decode(b).decode('utf-8').split('-')
            text[1] = str(int(text[1]) + 1)
            text = '-'.join(*[text])
        with open(self.filename, 'wb') as f:
            b = base64.b64encode(text.encode('utf-8'))
            f.write(b)
        return

    def register(self):
        if  not Path(self.filename).exists():
            self.initiate()
        with open(self.filename, 'rb') as f:
            b = f.read()
            text = base64.b64decode(b).decode('utf-8').split('-')
            text[0] = '1'
            text = '-'.join([*text])
        with open(self.filename, 'wb') as f:
            b = base64.b64encode(text.encode('utf-8'))
            f.write(b)

    def serial_number(self, serial):
        import urllib.request
        response = urllib.request.urlopen(self.gist_url)
        data = response.read()      # a `bytes` object
        text = data.decode('utf-8')
        return serial in text

def internet(host="8.8.8.8", port=53, timeout=3):
    import socket
    try:
        socket.setdefaulttimeout(timeout)
        socket.socket(socket.AF_INET, socket.SOCK_STREAM).connect((host, port))
        return True
    except Exception as ex:
        #         print(ex.message)
        return False