new script

check_versions.py

+#!/usr/bin/env python
+
+# Check software versions on the target
+
+__author__ = "Elphel"
+__copyright__ = "Copyright 2016, Elphel, Inc."
+__license__ = "GPL"
+__version__ = "3.0+"
+__maintainer__ = "Oleg K Dzhimiev"
+__email__ = "oleg@elphel.com"
+__status__ = "Development"
+
+import json
+import os
+import subprocess
+import sys
+    
+#http://stackoverflow.com/questions/287871/print-in-terminal-with-colors-using-python
+class bcolors:
+  HEADER = '\033[95m'
+  OKBLUE = '\033[94m'
+  OKGREEN = '\033[92m'
+  WARNING = '\033[38;5;214m'
+  FAIL = '\033[91m'
+  ENDC = '\033[0m'
+  BOLD = '\033[1m'
+  BOLDWHITE = '\033[1;37m'
+  UNDERLINE = '\033[4m'
+
+def shout(cmd):
+  subprocess.call(cmd,shell=True)
+
+def command_over_ssh(addr,command):
+    cmd = "ssh "+addr+" "+command
+    print("cmd: "+cmd)
+    try:
+      ret = subprocess.check_output(cmd,stderr=subprocess.STDOUT,shell=True)
+    except subprocess.CalledProcessError:
+      raise Exception("ssh to target requires access by public key. Run: \033[1;37mssh-copy-id "+addr+"\033[0m")
+    return ret.strip() 
+
+def read_local_conf(conf_file,pattern):
+  ret = "0"
+  if os.path.isfile(conf_file):
+    with open(conf_file,"r") as f:
+      lines = f.readlines()
+      for line in lines:
+        test = line.find(pattern)
+        if test!=-1:
+          ret = line.split("=")[1].strip().strip("\"")
+  return ret
+
+def get_versions_from_target(addr,tdir):
+  # print remote package list
+  tmp_str = command_over_ssh(addr,"'ls "+tdir+"'")
+  
+  remote_list = []
+  
+  tmp_list = tmp_str.split()
+  for elem in tmp_list:
+    remote_list.append([elem,command_over_ssh(addr,"'cat "+tdir+"/"+elem+"'")])
+  
+  return remote_list
+
+def get_version_from_git(path,vfile):
+  print(path)
+  cwd = os.getcwd()
+  os.chdir(cwd+"/"+path)
+  
+  p0=""
+  p1=""
+  
+  if os.path.isfile(vfile):
+    #PE.PV
+    
+    f=open(vfile)
+    for line in f:
+      line = line.strip()
+      if (line[0]!="#"):
+            break  
+    p0 = line
+    #PR
+    
+    cmd = "git rev-list --count $(git log -1 --pretty=format:\"%H\" "+vfile+")..HEAD"
+    p1 = subprocess.check_output(cmd,stderr=subprocess.STDOUT,shell=True)
+    p1 = p1.strip()
+        
+  else:
+    print(vfile+" file is missing in the project")
+    
+  os.chdir(cwd)
+  return p0+"."+p1
+
+def deep_analysis(local,remote):
+  print("\nVersion analysis")
+  
+  print("\n"+bcolors.BOLDWHITE+"{:<24}".format("Project")+"{:<16}".format("Local GIT")+"{:<16}".format("Target Version")+bcolors.ENDC)
+  
+  for pl,vl in local:
+    recstr = "{:<24}".format(pl)+"{:<16}".format(vl)
+    prfound = False
+    for pr,vr in remote:
+      if pl==pr:
+        prfound = True
+        recstr = recstr+"{:<16}".format(vr)
+        if vl==vr:
+          recstr = bcolors.OKGREEN+recstr+bcolors.ENDC
+        else:
+          recstr = bcolors.FAIL+recstr+bcolors.ENDC
+      if not prfound:
+        recstr = bcolors.WARNING+recstr+bcolors.ENDC
+          
+    print(recstr)
+  print("")
+
+usage = """Usage example:
+    {0}{1} root@192.168.0.9{2}, where
+        192.168.0.9 - target ip address
+        root        - target user
+""".format(bcolors.BOLDWHITE,sys.argv[0],bcolors.ENDC)
+
+# hardcoded
+user = ""
+ip = ""
+local_conf = "poky/build/conf/local.conf"
+target_dir = "/etc/elphel393/packages"
+local_project_list = "projects.json"
+local_dirs = ["rootfs-elphel","fpga-elphel","linux-elphel"]
+git_vfile = "VERSION"
+project_prefix = "elphel-"
+
+if len(sys.argv)>1:
+  rootip = sys.argv[1].split("@")
+  if len(rootip)>1:
+    user,ip = rootip
+  else:
+    ip = rootip[0]
+else:
+  if (os.path.isfile(local_conf)):
+    user   = read_local_conf(local_conf,"REMOTE_USER")
+    ip = read_local_conf(local_conf,"REMOTE_IP")
+    print("NOTE: user and ip are taken from "+local_conf)
+    print(usage)
+  
+  if user=="" or ip=="":
+    raise Exception(usage)
+
+print("Software/firmware versions check for target "+bcolors.BOLDWHITE+user+"@"+ip+bcolors.ENDC)
+
+print(bcolors.BOLDWHITE+"=== Read versions from the target ==="+bcolors.ENDC)
+target_list = get_versions_from_target(user+"@"+ip,target_dir)
+
+print(target_list)
+
+print(bcolors.BOLDWHITE+"=== Read local versions ==="+bcolors.ENDC)
+with open(local_project_list) as data_file:
+  Projects = json.load(data_file)
+
+local_list = []
+
+for p,v in Projects.items():
+  if p in local_dirs:
+    if isinstance(v,dict):
+      for k,l in v.items():
+        tmp = get_version_from_git(p+"/"+k,git_vfile)
+        if k.find(project_prefix)==0:
+          name = k[len(project_prefix):]
+        else:
+          name = k
+        local_list.append([name.encode('ascii','ignore'),tmp])
+    elif isinstance(v,list):
+      tmp = get_version_from_git(p,git_vfile)
+      local_list.append([p.encode('ascii','ignore'),tmp])
+    else:
+      raise Exception("Unknown error")
+
+print(local_list)
+
+deep_analysis(local_list,target_list)
+
+