[ViewVC] Annotation of: cvsroot/UserCode/OSUT3Analysis/DBTools/scripts/osucrab

#!/usr/bin/env perl

use strict;
use Mysql;

sub getWorkingDir;
sub dbUpdate;
sub dbStatus;
sub dbReport;
sub dbInsert;
sub addSlashes;
sub getOption;

our $db = Mysql->connect ("cmshead.mps.ohio-state.edu", "ntuple", "osuT3User") or die "Failed to connect to Tier 3, stopped";

my $arg = join (" ", @ARGV);
if ($arg =~ m/-create/)
  {
    my $crabCfgName = $arg;
    $crabCfgName =~ s/.*-cfg  *([^ \t\n]*).*/$1/;
    open (CRAB_CFG, "<$crabCfgName");
    my @crabCfg = <CRAB_CFG>;
    close (CRAB_CFG);
    my $dataset = getOption (\@crabCfg, "datasetpath");
    my $psetName = getOption (\@crabCfg, "pset");
    my $jsonFileName = getOption (\@crabCfg, "lumi_mask");
    my $submissionLogName = getOption (\@crabCfg, "ui_working_dir");
    $submissionLogName .= "/log/crab.log";

    print "Ntuple format (BEAN): ";
    my $format = <STDIN>;
    $format =~ s/\n//g;
    $format = "BEAN" if !$format;
    print "Comment for the ntuple database: ";
    my $comment = <STDIN>;
    $comment =~ s/\n//g;

    system ("crab $arg");

    my $status = "created";
    $status = "submitted" if $arg =~ m/-submit/;
    my $results = dbInsert ($dataset, "$ENV{'USER'}\@$ENV{'HOSTNAME'}", $format, "", 0, 0, $status, $comment, $psetName, $crabCfgName, $jsonFileName, $submissionLogName);
  }
elsif ($arg =~ m/-submit/)
  {
    my $workingDir = getWorkingDir (\@ARGV);
    my $id = -1;
    if (-e "$workingDir/log/.osuID")
      {
        open (ID_FILE, "<$workingDir/log/.osuID");
        $id = <ID_FILE>;
        close (ID_FILE);
        $id =~ s/(.*)\n/$1/;
      }
    my $submissionLogName = $workingDir . "/log/crab.log";

    system ("crab $arg");

    my $results = dbUpdate ($id, "submitted", $submissionLogName, "$ENV{'USER'}\@$ENV{'HOSTNAME'}") if $id > -1;
  }
elsif ($arg =~ m/-report/)
  {
    my $workingDir = getWorkingDir (\@ARGV);
    my $id = -1;
    if (-e "$workingDir/log/.osuID")
      {
        open (ID_FILE, "<$workingDir/log/.osuID");
        $id = <ID_FILE>;
        close (ID_FILE);
        $id =~ s/(.*)\n/$1/;
      }
    my $lumiSummaryName = $workingDir . "/res/lumiSummary.json";

    system ("crab $arg");

    my $results = dbReport ($id, "complete", $lumiSummaryName, "$ENV{'USER'}\@$ENV{'HOSTNAME'}") if $id > -1;
  }
elsif ($arg =~ m/-status/)
  {
    my $workingDir = getWorkingDir (\@ARGV);
    my $id = -1;
    if (-e "$workingDir/log/.osuID")
      {
        open (ID_FILE, "<$workingDir/log/.osuID");
        $id = <ID_FILE>;
        close (ID_FILE);
        $id =~ s/(.*)\n/$1/;
      }
    my $currentStatusName = "$workingDir/log/.currentStatus";
    unlink ($currentStatusName) if -e $currentStatusName;
    system ("crab $arg | tee $currentStatusName");
    my $results = dbStatus ($id, $currentStatusName, "$ENV{'USER'}\@$ENV{'HOSTNAME'}") if $id > -1;
    unlink ($currentStatusName);
  }
else
  {
    system ("crab $arg");
  }

sub
getWorkingDir
{
  my $argv = shift;

  my $workingDir;
  my $nextToLast = 0;
  foreach my $argument (@$argv)
    {
      if ($nextToLast)
        {
          $workingDir = $argument;
          last;
        }
      $nextToLast = 1 if ($argument eq "-c" || $argument eq "-continue");
    }

  return $workingDir;
}

sub
getOption
{
  my $file = shift;
  my $option = shift;

  my $optionValue;
  foreach my $line (@$file)
    {
      next if (!($line =~ m/$option  *=  *[^ \t\n]*/));
      $optionValue = $line;
      $optionValue =~ s/.*$option  *=  *([^ \t\n]*).*\n/$1/;
    }

  return $optionValue;
}

sub
addSlashes
{
  my $string = shift;

  $string =~ s/\\/\\\\/g;
  $string =~ s/'/\\'/g;
  $string =~ s/"/\\"/g;
  $string =~ s/\\0/\\\\0/g;

  return $string;
}

sub
dbInsert
{
  my $dataset = shift;
  my $userAndHost = shift;
  my $format = shift;
  my $location = shift;
  my $nFiles = shift;
  my $sizeInGB = shift;
  my $status = shift;
  my $comment = shift;
  my $psetName = shift;
  my $crabCfgName = shift;
  my $jsonFileName = shift;
  my $submissionLogName = shift;

  my $user = $userAndHost;
  $user =~ s/@.*$//;

  open (PY_CONFIG, "<$psetName");
  my @pset = <PY_CONFIG>;
  close (PY_CONFIG);
  my $pset = join ("", @pset);
  open (CRAB_CONFIG, "<$crabCfgName");
  my @crabCfg = <CRAB_CONFIG>;
  close (CRAB_CONFIG);
  my $crabCfg = join ("", @crabCfg);
  open (JSON_FILE, "<$jsonFileName");
  my @jsonFile = <JSON_FILE>;
  close (JSON_FILE);
  my $jsonFile = join ("", @jsonFile);
  open (SUBMISSION_LOG, "<$submissionLogName");
  my @submissionLog = <SUBMISSION_LOG>;
  close (SUBMISSION_LOG);
  my $submissionLog = join ("", @submissionLog);

  $dataset = addSlashes ($dataset);
  $user = addSlashes ($user);
  $format = addSlashes ($format);
  $pset = addSlashes ($pset);
  $crabCfg = addSlashes ($crabCfg);
  $jsonFile = addSlashes ($jsonFile);
  $submissionLog = addSlashes ($submissionLog);
  $location = addSlashes ($location);
  $status = addSlashes ($status);
  $comment = addSlashes ($comment);

  my $results;
  my $query = "select max(id) from ntuple";
  $db->selectdb ("ntuple");
  $results = $db->query ($query);
  my @row = $results->fetchrow ();
  my $id = 1;
  $id = $row[0] + 1 if $results->numrows ();

  my $workingDir = $submissionLogName;
  $workingDir =~ s/(.*)\/log\/crab\.log/$1/;
  open (ID_FILE, ">$workingDir/log/.osuID");
  print ID_FILE "$id\n";
  close (ID_FILE);

  $query = "insert into ntuple (id, dataset, creationTime, lastUpdateTime, lastUpdateUser, user, format, location, nFiles, sizeInGB, status, comment, pset, crabCfg, jsonFile, submissionLog) values ($id, '$dataset', now(), now(), '$userAndHost', '$user', '$format', '$location', $nFiles, $sizeInGB, '$status', '$comment', '$pset', '$crabCfg', '$jsonFile', '$submissionLog')";
  $results = $db->query ($query);

  return $results;
}

sub
dbUpdate
{
  my $id = shift;
  my $status = shift;
  my $submissionLogName = shift;
  my $user = shift;

  open (SUBMISSION_LOG, "<$submissionLogName");
  my @submissionLog = <SUBMISSION_LOG>;
  close (SUBMISSION_LOG);
  my $submissionLog = join ("", @submissionLog);

  $submissionLog = addSlashes ($submissionLog);
  $status = addSlashes ($status);
  $user = addSlashes ($user);

  my $results;
  my $query = "update ntuple set lastUpdateTime=now(), lastUpdateUser='$user', status='$status', submissionLog='$submissionLog' where id=$id";
  $db->selectdb ("ntuple");
  $results = $db->query ($query);

  return $results;
}

sub
dbStatus
{
  my $id = shift;
  my $currentStatusName = shift;
  my $user = shift;

  open (CURRENT_STATUS, "<$currentStatusName");
  my @currentStatus = <CURRENT_STATUS>;
  close (CURRENT_STATUS);
  my $currentStatus = join ("", @currentStatus);

  $currentStatus = addSlashes ($currentStatus);
  $user = addSlashes ($user);

  my $results;
  my $query = "update ntuple set lastUpdateTime=now(), lastUpdateUser='$user', currentStatus='$currentStatus' where id=$id";
  $db->selectdb ("ntuple");
  $results = $db->query ($query);

  return $results;
}

sub
dbReport
{
  my $id = shift;
  my $status = shift;
  my $lumiSummaryName = shift;
  my $user = shift;

  open (LUMI_SUMMARY, "<$lumiSummaryName");
  my @lumiSummary = <LUMI_SUMMARY>;
  close (LUMI_SUMMARY);
  my $lumiSummary = join ("", @lumiSummary);

  $lumiSummary = addSlashes ($lumiSummary);
  $status = addSlashes ($status);
  $user = addSlashes ($user);

  my $results;
  my $query = "update ntuple set lastUpdateTime=now(), lastUpdateUser='$user', status='$status', lumiSummary='$lumiSummary' where id=$id";
  $db->selectdb ("ntuple");
  $results = $db->query ($query);

  return $results;
}
Revision:	1.6
Committed:	Fri Nov 30 18:21:00 2012 UTC (12 years, 5 months ago) by ahart
Branch:	MAIN
Changes since 1.5:	+5 -2 lines
Log Message:	Only append the hostname for the lastUpdateUser field.
#	User	Rev	Content
1	ahart	1.1	#!/usr/bin/env perl
2
3			use strict;
4			use Mysql;
5
6			sub getWorkingDir;
7			sub dbUpdate;
8	ahart	1.4	sub dbStatus;
9	ahart	1.1	sub dbReport;
10			sub dbInsert;
11			sub addSlashes;
12			sub getOption;
13
14	ahart	1.2	our $db = Mysql->connect ("cmshead.mps.ohio-state.edu", "ntuple", "osuT3User") or die "Failed to connect to Tier 3, stopped";
15	ahart	1.1
16			my $arg = join (" ", @ARGV);
17			if ($arg =~ m/-create/)
18			{
19			my $crabCfgName = $arg;
20			$crabCfgName =~ s/.-cfg ([^ \t\n])./$1/;
21			open (CRAB_CFG, "<$crabCfgName");
22			my @crabCfg = <CRAB_CFG>;
23			close (CRAB_CFG);
24			my $dataset = getOption (\@crabCfg, "datasetpath");
25			my $psetName = getOption (\@crabCfg, "pset");
26			my $jsonFileName = getOption (\@crabCfg, "lumi_mask");
27			my $submissionLogName = getOption (\@crabCfg, "ui_working_dir");
28			$submissionLogName .= "/log/crab.log";
29
30			print "Ntuple format (BEAN): ";
31			my $format = <STDIN>;
32			$format =~ s/\n//g;
33			$format = "BEAN" if !$format;
34			print "Comment for the ntuple database: ";
35			my $comment = <STDIN>;
36			$comment =~ s/\n//g;
37
38			system ("crab $arg");
39
40			my $status = "created";
41			$status = "submitted" if $arg =~ m/-submit/;
42	ahart	1.5	my $results = dbInsert ($dataset, "$ENV{'USER'}\@$ENV{'HOSTNAME'}", $format, "", 0, 0, $status, $comment, $psetName, $crabCfgName, $jsonFileName, $submissionLogName);
43	ahart	1.1	}
44			elsif ($arg =~ m/-submit/)
45			{
46			my $workingDir = getWorkingDir (\@ARGV);
47	ahart	1.3	my $id = -1;
48			if (-e "$workingDir/log/.osuID")
49			{
50			open (ID_FILE, "<$workingDir/log/.osuID");
51			$id = <ID_FILE>;
52			close (ID_FILE);
53			$id =~ s/(.*)\n/$1/;
54			}
55	ahart	1.1	my $submissionLogName = $workingDir . "/log/crab.log";
56
57			system ("crab $arg");
58
59	ahart	1.5	my $results = dbUpdate ($id, "submitted", $submissionLogName, "$ENV{'USER'}\@$ENV{'HOSTNAME'}") if $id > -1;
60	ahart	1.1	}
61			elsif ($arg =~ m/-report/)
62			{
63			my $workingDir = getWorkingDir (\@ARGV);
64	ahart	1.3	my $id = -1;
65			if (-e "$workingDir/log/.osuID")
66			{
67			open (ID_FILE, "<$workingDir/log/.osuID");
68	ahart	1.4	$id = <ID_FILE>;
69	ahart	1.3	close (ID_FILE);
70			$id =~ s/(.*)\n/$1/;
71			}
72	ahart	1.1	my $lumiSummaryName = $workingDir . "/res/lumiSummary.json";
73
74			system ("crab $arg");
75
76	ahart	1.5	my $results = dbReport ($id, "complete", $lumiSummaryName, "$ENV{'USER'}\@$ENV{'HOSTNAME'}") if $id > -1;
77	ahart	1.4	}
78			elsif ($arg =~ m/-status/)
79			{
80			my $workingDir = getWorkingDir (\@ARGV);
81			my $id = -1;
82			if (-e "$workingDir/log/.osuID")
83			{
84			open (ID_FILE, "<$workingDir/log/.osuID");
85			$id = <ID_FILE>;
86			close (ID_FILE);
87			$id =~ s/(.*)\n/$1/;
88			}
89			my $currentStatusName = "$workingDir/log/.currentStatus";
90			unlink ($currentStatusName) if -e $currentStatusName;
91			system ("crab $arg \| tee $currentStatusName");
92	ahart	1.5	my $results = dbStatus ($id, $currentStatusName, "$ENV{'USER'}\@$ENV{'HOSTNAME'}") if $id > -1;
93	ahart	1.4	unlink ($currentStatusName);
94	ahart	1.1	}
95			else
96			{
97			system ("crab $arg");
98			}
99
100			sub
101			getWorkingDir
102			{
103			my $argv = shift;
104
105			my $workingDir;
106			my $nextToLast = 0;
107			foreach my $argument (@$argv)
108			{
109			if ($nextToLast)
110			{
111			$workingDir = $argument;
112			last;
113			}
114			$nextToLast = 1 if ($argument eq "-c" \|\| $argument eq "-continue");
115			}
116
117			return $workingDir;
118			}
119
120			sub
121			getOption
122			{
123			my $file = shift;
124			my $option = shift;
125
126			my $optionValue;
127			foreach my $line (@$file)
128			{
129			next if (!($line =~ m/$option = [^ \t\n]*/));
130			$optionValue = $line;
131			$optionValue =~ s/.$option = ([^ \t\n]).*\n/$1/;
132			}
133
134			return $optionValue;
135			}
136
137			sub
138			addSlashes
139			{
140			my $string = shift;
141
142			$string =~ s/\\/\\\\/g;
143			$string =~ s/'/\\'/g;
144			$string =~ s/"/\\"/g;
145			$string =~ s/\\0/\\\\0/g;
146
147			return $string;
148			}
149
150			sub
151			dbInsert
152			{
153			my $dataset = shift;
154	ahart	1.6	my $userAndHost = shift;
155	ahart	1.1	my $format = shift;
156			my $location = shift;
157			my $nFiles = shift;
158			my $sizeInGB = shift;
159			my $status = shift;
160			my $comment = shift;
161			my $psetName = shift;
162			my $crabCfgName = shift;
163			my $jsonFileName = shift;
164			my $submissionLogName = shift;
165
166	ahart	1.6	my $user = $userAndHost;
167			$user =~ s/@.*$//;
168
169	ahart	1.1	open (PY_CONFIG, "<$psetName");
170			my @pset = <PY_CONFIG>;
171			close (PY_CONFIG);
172			my $pset = join ("", @pset);
173			open (CRAB_CONFIG, "<$crabCfgName");
174			my @crabCfg = <CRAB_CONFIG>;
175			close (CRAB_CONFIG);
176			my $crabCfg = join ("", @crabCfg);
177			open (JSON_FILE, "<$jsonFileName");
178			my @jsonFile = <JSON_FILE>;
179			close (JSON_FILE);
180			my $jsonFile = join ("", @jsonFile);
181			open (SUBMISSION_LOG, "<$submissionLogName");
182			my @submissionLog = <SUBMISSION_LOG>;
183			close (SUBMISSION_LOG);
184			my $submissionLog = join ("", @submissionLog);
185
186			$dataset = addSlashes ($dataset);
187			$user = addSlashes ($user);
188			$format = addSlashes ($format);
189			$pset = addSlashes ($pset);
190			$crabCfg = addSlashes ($crabCfg);
191			$jsonFile = addSlashes ($jsonFile);
192			$submissionLog = addSlashes ($submissionLog);
193			$location = addSlashes ($location);
194			$status = addSlashes ($status);
195			$comment = addSlashes ($comment);
196
197			my $results;
198			my $query = "select max(id) from ntuple";
199			$db->selectdb ("ntuple");
200			$results = $db->query ($query);
201			my @row = $results->fetchrow ();
202			my $id = 1;
203			$id = $row[0] + 1 if $results->numrows ();
204
205			my $workingDir = $submissionLogName;
206			$workingDir =~ s/(.*)\/log\/crab\.log/$1/;
207			open (ID_FILE, ">$workingDir/log/.osuID");
208			print ID_FILE "$id\n";
209			close (ID_FILE);
210
211	ahart	1.6	$query = "insert into ntuple (id, dataset, creationTime, lastUpdateTime, lastUpdateUser, user, format, location, nFiles, sizeInGB, status, comment, pset, crabCfg, jsonFile, submissionLog) values ($id, '$dataset', now(), now(), '$userAndHost', '$user', '$format', '$location', $nFiles, $sizeInGB, '$status', '$comment', '$pset', '$crabCfg', '$jsonFile', '$submissionLog')";
212	ahart	1.1	$results = $db->query ($query);
213
214			return $results;
215			}
216
217			sub
218			dbUpdate
219			{
220			my $id = shift;
221			my $status = shift;
222			my $submissionLogName = shift;
223	ahart	1.4	my $user = shift;
224	ahart	1.1
225			open (SUBMISSION_LOG, "<$submissionLogName");
226			my @submissionLog = <SUBMISSION_LOG>;
227			close (SUBMISSION_LOG);
228			my $submissionLog = join ("", @submissionLog);
229
230			$submissionLog = addSlashes ($submissionLog);
231			$status = addSlashes ($status);
232	ahart	1.4	$user = addSlashes ($user);
233	ahart	1.1
234			my $results;
235	ahart	1.4	my $query = "update ntuple set lastUpdateTime=now(), lastUpdateUser='$user', status='$status', submissionLog='$submissionLog' where id=$id";
236			$db->selectdb ("ntuple");
237			$results = $db->query ($query);
238
239			return $results;
240			}
241
242			sub
243			dbStatus
244			{
245			my $id = shift;
246			my $currentStatusName = shift;
247			my $user = shift;
248
249			open (CURRENT_STATUS, "<$currentStatusName");
250			my @currentStatus = <CURRENT_STATUS>;
251			close (CURRENT_STATUS);
252			my $currentStatus = join ("", @currentStatus);
253
254			$currentStatus = addSlashes ($currentStatus);
255			$user = addSlashes ($user);
256
257			my $results;
258			my $query = "update ntuple set lastUpdateTime=now(), lastUpdateUser='$user', currentStatus='$currentStatus' where id=$id";
259	ahart	1.1	$db->selectdb ("ntuple");
260			$results = $db->query ($query);
261
262			return $results;
263			}
264
265			sub
266			dbReport
267			{
268			my $id = shift;
269			my $status = shift;
270			my $lumiSummaryName = shift;
271	ahart	1.4	my $user = shift;
272	ahart	1.1
273			open (LUMI_SUMMARY, "<$lumiSummaryName");
274			my @lumiSummary = <LUMI_SUMMARY>;
275			close (LUMI_SUMMARY);
276			my $lumiSummary = join ("", @lumiSummary);
277
278			$lumiSummary = addSlashes ($lumiSummary);
279			$status = addSlashes ($status);
280	ahart	1.4	$user = addSlashes ($user);
281	ahart	1.1
282			my $results;
283	ahart	1.4	my $query = "update ntuple set lastUpdateTime=now(), lastUpdateUser='$user', status='$status', lumiSummary='$lumiSummary' where id=$id";
284	ahart	1.1	$db->selectdb ("ntuple");
285			$results = $db->query ($query);
286
287			return $results;
288			}