*

2025-05-13 01:34:53 +03:00
parent 427735e23d
commit 83f3f1c7d4
945 changed files with 633484 additions and 0 deletions
@@ -0,0 +1,98 @@
+
+import re
+import subprocess
+import os
+import sys
+import platform
+
+def version_str_to_tuple(version_string):
+    """
+    Converts a version string to a tuple of the form
+    (major, minor, revision, prerelease)
+
+    Example: "fw-v0.3.6-23" => (0, 3, 6, True)
+
+    If version_string does not match the pattern above, this function throws an
+    Exception.
+    """
+    regex=r'.*v([0-9]+)\.([0-9]+)\.([0-9]+)(.*)'
+    if not re.match(regex, version_string):
+        raise Exception()
+    return (int(re.sub(regex, r"\1", version_string)),
+            int(re.sub(regex, r"\2", version_string)),
+            int(re.sub(regex, r"\3", version_string)),
+            (re.sub(regex, r"\4", version_string) != ""))
+
+
+def get_version_from_git():
+    script_dir = os.path.dirname(os.path.realpath(__file__))
+    try:
+        # Determine the current git commit version
+        git_tag = subprocess.check_output(["git", "describe", "--always", "--tags", "--dirty=*"],
+            cwd=script_dir)
+        git_tag = git_tag.decode(sys.stdout.encoding).rstrip('\n')
+
+        (major, minor, revision, is_prerelease) = version_str_to_tuple(git_tag)
+
+        # if is_prerelease:
+        #     revision += 1
+        return git_tag, major, minor, revision, is_prerelease
+
+    except Exception as ex:
+        print(ex)
+        return "[unknown version]", 0, 0, 0, 1
+
+def get_version_str(git_only=False, is_post_release=False, bump_rev=False, release_override=False):
+    """
+    Returns the versions of the tools
+    If git_only is true, the version.txt file is ignored even
+    if it is present.
+    """
+    script_dir = os.path.dirname(os.path.realpath(__file__))
+
+    # Try to read the version.txt file that is generated during
+    # the packaging step
+    version_file_path = os.path.join(script_dir, 'version.txt')
+    if os.path.exists(version_file_path) and git_only == False:
+        with open(version_file_path) as version_file:
+            return version_file.readline().rstrip('\n')
+    
+    _, major, minor, revision, unreleased = get_version_from_git()
+    if bump_rev:
+        revision += 1
+    version = '{}.{}.{}'.format(major, minor, revision)
+    if is_post_release:
+        version += ".post"
+    elif not release_override and unreleased:
+        version += ".dev"
+    return version
+
+if __name__ == '__main__':
+    import argparse
+    parser = argparse.ArgumentParser(description='Version Dump\n')
+    parser.add_argument("--output", type=argparse.FileType('w'), default='-',
+                        help="C header output file")
+    
+    args = parser.parse_args()
+
+    git_name, major, minor, revision, unreleased = get_version_from_git()
+    print('Firmware version {}.{}.{}{} ({})'.format(
+        major, minor, revision, '-dev' if unreleased else '',
+        git_name))
+    #args.output.write('const unsigned char fw_version = "{}"\n'.format(git_name))
+    args.output.write('const unsigned char fw_version_major_ = {};\n'.format(major))
+    args.output.write('const unsigned char fw_version_minor_ = {};\n'.format(minor))
+    args.output.write('const unsigned char fw_version_revision_ = {};\n'.format(revision))
+    args.output.write('const unsigned char fw_version_unreleased_ = {};\n'.format(1 if unreleased else 0))
+
+def setup_udev_rules(logger):
+  if platform.system() != 'Linux':
+      if logger: logger.error("This command only makes sense on Linux")
+      return
+  if os.getuid() != 0:
+      if logger: logger.warn("you should run this as root, otherwise it will probably not work")
+  with open('/etc/udev/rules.d/91-odrive.rules', 'w') as file:
+      file.write('SUBSYSTEM=="usb", ATTR{idVendor}=="1209", ATTR{idProduct}=="0d3[0-9]", MODE="0666", ENV{ID_MM_DEVICE_IGNORE}="1"\n')
+  subprocess.check_call(["udevadm", "control", "--reload-rules"])
+  subprocess.check_call(["udevadm", "trigger"])
+  if logger: logger.info('udev rules configured successfully')