X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=pubmed_search;h=c0d15ce50afb745f3fb2a6f3b781a21e131dacaf;hb=3d5241a316e3ff729b19b878b0841558120f75e9;hp=b3367d934497f13391a79cbadb62391aaabc9d05;hpb=d854925a60b80c6866f148dd70e717da6186ce32;p=bin.git

diff --git a/pubmed_search b/pubmed_search
index b3367d9..c0d15ce 100755
--- a/pubmed_search
+++ b/pubmed_search
@@ -21,7 +21,7 @@ use Text::Wrap;
 
 =head1 NAME
 
-pubmed_search - 
+pubmed_search - Search for articles on pubmed
 
 =head1 SYNOPSIS
 
@@ -59,16 +59,19 @@ Display this manual.
 use vars qw($DEBUG);
 
 my %options = (debug           => 0,
-	       help            => 0,
-	       man             => 0,
-	       color           => 1,
-	       org_mode        => 0,
-	       );
+               help            => 0,
+               man             => 0,
+               color           => 1,
+               org_mode        => 0,
+               abstract        => 1,
+              );
 
 GetOptions(\%options,
-	   'color|c!',
-	   'org_mode|org-mode',
-	   'debug|d+','help|h|?','man|m');
+           'color|c!',
+           'org_mode|org-mode',
+           'pmid_only|pmid-only',
+           'abstract|a!',
+           'debug|d+','help|h|?','man|m');
 
 pod2usage() if $options{help};
 pod2usage({verbose=>2}) if $options{man};
@@ -83,33 +86,52 @@ if (not @ARGV) {
 pod2usage(join("\n",@USAGE_ERRORS)) if @USAGE_ERRORS;
 
 
-my $search = Bio::DB::EUtilities->new(-eutil => 'esearch',
-				       -email => 'don@donarmstrong.com',
-				       -db    => 'pubmed',
-				       -term => join(' ',@ARGV),
-				       -retmax => 1000,
-				      );
-my @ids = $search->get_ids();
+my $result =
+    Bio::DB::EUtilities->new(-email => 'don@donarmstrong.com',
+                             -db    => 'pubmed',
+                             -term => join(' ',@ARGV),
+                             -retmax => 1000,
+                             -eutil  => 'esearch',
+                            );
+my @ids;
+eval {
+    # this warns for everything, so hide the warning.
+    local $SIG{__WARN__} = sub {};
+    @ids = $result->get_ids('pubmed')
+};
+if (not @ids) {
+    print "No results\n";
+    exit;
+}
+if (@ids > 0 and ref($ids[0])) {
+    @ids = @{$ids[0]};
+}
 if ($options{org_mode}) {
     print "* Pubmed search results for ".join(' ',@ARGV)." (".scalar(@ids).")\n";
     print "  + ";
+} elsif ($options{pmid_only}) {
+    print map{qq($_\n)} @ids;
+    exit 0;
 }
 print scalar(@ids)." results:\n";
 exit 0 unless @ids;
-my $esummary = Bio::DB::EUtilities->new(-eutil => 'efetch',
-					-email => 'don@donarmstrong.com',
-					-db    => 'pubmed',
-					-id  => \@ids
-				       );
+my $raw_xml = Bio::DB::EUtilities->new(-email => 'don@donarmstrong.com',
+                                       -db    => 'pubmed',
+                                       -id  => \@ids,
+                                       -eutil => 'efetch',
+                                      );
 use XML::LibXML;
-my $xml = XML::LibXML->load_xml(string => $esummary->get_Response()->content());
-for my $article ($xml->findnodes('PubmedArticleSet/PubmedArticle/MedlineCitation')) {
-    # print $article->toString;
-    my ($pmid) = $article->findnodes('./PMID');
-    my ($title) = $article->findnodes('./Article/ArticleTitle');
-    my ($abstract) = $article->findnodes('./Article/Abstract');
+my $xml = XML::LibXML->load_xml(string => $raw_xml->get_Response->content);
+print STDERR $xml->toString if $DEBUG;
+for my $article ($xml->findnodes(q{//*[local-name()='MedlineCitation']})) {
+    print STDERR $article->toString if $DEBUG;
+    my ($pmid) = $article->findnodes(q{./*[local-name()='PMID']});
+    my ($title) = $article->findnodes(q{./*[local-name()='Article']}.
+                                      q{/*[local-name()='ArticleTitle']});
+    my ($abstract) = $article->findnodes(q{./*[local-name()='Article']}.
+                                         q{/*[local-name()='Abstract']});
     if ($options{org_mode}) {
-	print "** PMID: ";
+        print "** PMID: ";
     }
     print BOLD GREEN if $options{color};
     print $pmid->textContent();
@@ -118,18 +140,18 @@ for my $article ($xml->findnodes('PubmedArticleSet/PubmedArticle/MedlineCitation
     print BOLD CYAN if $options{color};
     print encode_utf8($title->textContent())."\n";
     print RESET if $options{color};
-    print BOLD MAGENTA if $options{color};
-    if (defined $abstract) {
-	if ($options{org_mode}) {
-	    print "*** Abstract\n";
-	}
-	$abstract = $abstract->textContent();
-	$abstract =~ s/^\s*//mg;
-	$abstract =~ s/(.{,80})\s/$1\n/g;
-	$abstract = encode_utf8($abstract);
-	print wrap('','',$abstract);
-	print "\n\n";
-	print RESET if $options{color};
+    if (defined $abstract and $options{abstract}) {
+        print BOLD MAGENTA if $options{color};
+        if ($options{org_mode}) {
+            print "*** Abstract\n";
+        }
+        $abstract = $abstract->textContent();
+        $abstract =~ s/^\s*//mg;
+        $abstract =~ s/(.{,80})\s/$1\n/g;
+        $abstract = encode_utf8($abstract);
+        print wrap('','',$abstract);
+        print "\n\n";
+        print RESET if $options{color};
     }
 }