def Upload(self, ss_key, ws_name):
"""Upload |_csv_table| to the given Google Spreadsheet.
The spreadsheet is identified the spreadsheet key |ss_key|.
The worksheet within that spreadsheet is identified by the
worksheet name |ws_name|.
"""
if self._scomm:
self._scomm.SetCurrentWorksheet(ws_name)
else:
self._scomm = gdata_lib.SpreadsheetComm()
self._scomm.Connect(self._creds, ss_key, ws_name,
source='Upload Package Status')
oper.Notice('Caching rows for worksheet %r.' % self._scomm.ws_name)
self._ss_row_cache = self._scomm.GetRowCacheByCol(self.SS_ID_COL)
oper.Notice('Uploading changes to worksheet "%s" of spreadsheet "%s" now.' %
(self._scomm.ws_name, self._scomm.ss_key))
oper.Info('Details by package: S=Same, C=Changed, A=Added, D=Deleted')
rows_unchanged, rows_updated, rows_inserted = self._UploadChangedRows()
rows_deleted, rows_with_owner_deleted = self._DeleteOldRows()
oper.Notice('Final row stats for worksheet "%s"'
': %d changed, %d added, %d deleted, %d same.' %
(self._scomm.ws_name, rows_updated, rows_inserted,
rows_deleted, rows_unchanged))
if rows_with_owner_deleted:
oper.Warning('%d rows with owner entry deleted, see above warnings.' %
rows_with_owner_deleted)
else:
oper.Notice('No rows with owner entry were deleted.')
def NewScomm(self, gd_client=None, connect=True):
"""Return a non-mocked SpreadsheetComm."""
scomm = gdata_lib.SpreadsheetComm()
scomm.gd_client = gd_client
if connect:
scomm.ss_key = self.SS_KEY
scomm.ws_name = self.WS_NAME
scomm.ws_key = self.WS_KEY
else:
scomm.ss_key = None
scomm.ws_name = None
scomm.ws_key = None
return scomm
def main(argv):
"""Main function."""
parser = _CreateOptParser()
(options, _args) = parser.parse_args(argv)
oper.verbose = options.verbose
_CheckOptions(options)
ss_key = ups.TEST_SS_KEY if options.test_ss else ups.REAL_SS_KEY
# Prepare credentials for Docs and Tracker access.
creds = PrepareCreds(options.cred_file, options.token_file, options.email)
scomm = gdata_lib.SpreadsheetComm()
scomm.Connect(creds, ss_key, PKGS_WS_NAME, source='Sync Package Status')
tcomm = gdata_lib.TrackerComm()
tcomm.Connect(creds, PROJECT_NAME, source='Sync Package Status')
oper.Notice('Syncing between Tracker and spreadsheet %s' % ss_key)
syncer = Syncer(tcomm, scomm,
pretend=options.pretend, verbose=options.verbose)
if options.team:
syncer.SetTeamFilter(options.team)
if options.owner:
syncer.SetOwnerFilter(options.owner)
if options.default_owner:
syncer.SetDefaultOwner(options.default_owner)
try:
syncer.Sync()
except SyncError as ex:
oper.Die(str(ex))
# If --email, which is only effective when run interactively (because the
# password must be entered), give the option of saving to a creds file for
# next time.
if options.email and options.cred_file:
prompt = ('Do you want to save credentials for next time to %r?' %
options.cred_file)
if cros_build_lib.BooleanPrompt(prompt, False):
creds.StoreCreds(options.cred_file)
oper.Notice('Be sure to save the creds file to the same location'
' outside your chroot so it will also be used with'
' future chroots.')