#!/usr/bin/env python

"""This program is for updating the status of tapes in DataProtector.

Barcode readers generally act like PS2 or USB keyboards. They enter
whatever they zap, and then send a new-line character.

So this program reads data from stdin in that format. 

It expects a command-line argument which is the location to update those
tapes to have.

Almost no error checking is done.

It needs to run on a machine which has the cell console software on it,
and as a user who has at least the rights to run omnimm.

(c) Greg Baker 2010-2011 <gregb@ifost.org.au>

$Id: tape-zap.py 790 2011-12-14 01:17:57Z gregb $
"""

import os
import sys
import string
import subprocess

if len(sys.argv) == 1:
  sys.exit("Please supply a location you would like the tapes to be moved to")

dest_location = string.join(sys.argv[1:]," ")

print "Marking tapes as being in location: ",dest_location
# Perhaps I should check that this is a valid location?

if sys.platform != 'win32':
  omnimm = '/opt/omni/bin/omnimm'
else:
  omnimm = None
  for possibility in ["C:\\Program Files\\OmniBack\\bin\\omnimm.exe","D:\\Program Files\\Omniback\\bin\\omnimm.exe"]:
    if os.path.exists(possibility):
      omnimm = possibility
      break 
  if omnimm is None:
    sys.exit("Can't find omnimm program")

while 1:
  print "Zap or enter the label of the tape (x to exit): ",
  tape_name = sys.stdin.readline()
  if tape_name == "": break
  # Remove newline char
  tape_name = tape_name[:-1] 
  if tape_name == 'x' or tape_name == 'X': break
  # Assume that no-one puts " or funny chars into 
  # If we were sure to be on unix, I'd just use os.fork and os.exevc so
  # that no quoting was needed
  command = [omnimm,"-modify_medium",tape_name,tape_name,dest_location]
  result = subprocess.call(command)
  if result == 0:
    print "\a",
  else:
    print "\a\a"
    sys.stderr.write("Command failed: " + string.join(" ",command))