МХАТу пьеса о Сталине была необходима в преддверии празднования 60-летия вождя, которое готовились отмечать 21 декабря 1939 года. Правда, на самом деле Сталин родился не 9/21 декабря 1879 года, как неизменно писалось начиная с 1929 года во всех официальных биографических очерках, а 6/18 декабря 1878 года. В 1990 году была опубликована метрическая запись 1878 года Горийской соборной церкви: «6 декабря у жителей Гори крестьянина Виссариона Ивановича Джугашвили и его законной супруги Екатерины Габриеловны (оба православного вероисповедания) родился сын Иосиф». Скорее всего, дату своего рождения Сталин исказил без всякого умысла. Просто до революции, как правило, отмечали день ангела, а не день рождения, который часто и не запоминали. А год рождения полвека спустя Сталин мог и спутать, тем более что, по преданию, он сказал людям, писавшим его биографию: «Разве сейчас это важно – год рождения? Пишите как хотите». Так или иначе, дата 21 декабря 1879 года стала частью сталинского мифа и остается ею и сегодня. Такая же роль была уготована и будущей пьесе.
Н.А. Земской Булгаков признавался, что его интересовал образ молодого Сталина, романтического и живого юноши, только что включившегося в революционное движение. Вероятно, драматург надеялся, что ему не придется идти на компромисс с совестью, что молодой Сталин был одержим идеалами борьбы за справедливость и свободу против самодержавия и лишь годы спустя превратился в жестокого вождя большевиков. В ростовском журнале «Донская волна» Булгаков мог читать в 1919 году очерк о Сталине перебежчика полковника А.Л. Носовича (под псевдонимом «А. Черноморцев»), бывшего вместе со Сталиным на Царицынском фронте и отмечавшего его невероятную жестокость и большую силу воли. На очерк Носовича, вероятно, первый вообще очерк о Сталине, ссылался и Ворошилов в 1929 году в статье «Сталин и Красная Армия», выбирая при этом лишь позитивные отзывы о вожде. Конечно, никаких архивных материалов автор «Батума» так и не получил, да, скорее всего, он на них и не рассчитывал, если заявил Виленкину и Маркову, что сперва напишет пьесу, а потом пусть Немирович пишет письмо о материале. Если в 1936 году еще не было опубликовано почти никаких специальных материалов о молодом Сталине, то к сентябрю 1938 года, когда после разговора с мхатовцами Булгаков приступил к работе над пьесой, подобные материалы уже имелись.
Самые ранние наброски к пьесе, которая сегодня известна под именем «Батум», на столе Булгакова появились 16 января 1939 года. Первая редакция пьесы называлась «Пастырь». В ее основу была положена история Батумской рабочей демонстрации 8–9 марта 1902 года, организованной Сталиным. Главным источником послужила книга «Батумская демонстрация 1902 года», выпущенная Партиздатом в марте 1937 года и содержавшая документы и воспоминания, призванные возвеличить первые шаги вождя по руководству революционным движением в Закавказье. В качестве вариантов заглавия, кроме «Пастырь», Булгаков рассматривал «Бессмертие», «Битва», «Рождение славы», «Аргонавты», «Геракл», «Кормчий», «Юность штурмана», «Так было», «Кондор», «Комета зажглась», «Штурман вел корабль», «Молния», «Вставший из снега», «Штурман вел по звездам», «Юность командора», «Юный штурман», «Юность рулевого», «Поход аргонавтов», «Штурман шел по звездам», «Море штормит», «Когда начинался шторм», «Шторм грохотал», «Будет буря», «Мастер», «Штурман вел аргонавтов», «Комета пришла», «Как начиналась слава», «У огня», «Дело было в Батуми» и, наконец, «Батум». Поскольку действие пьесы разворачивается в черноморском порте и на месте древнегреческой колонии, Булгаков пробовал названия, связанные с морем или героями древнегреческих мифов – аргонавтами, Гераклом. В ряде названий, начиная с «Пастыря», фигурирует образ молодого Сталина, выступающего в пьесе в качестве положительного культурного героя мифа (иная трактовка по условиям цензуры была в принципе невозможна).
15 июня Булгаков подписал договор с МХАТом. Первая редакция пьесы была закончена в середине июля, а беловая редакция – к 27 июля, когда Булгаков прочел «Батум» партийной группе МХАТа. Это чтение описано Е.С. Булгаковой: «Слушали замечательно, после чтения очень долго, стоя, аплодировали. Потом высказыванья. Все очень хорошо. Калишьян в последней речи сказал, что театр должен ее поставить к 21 декабря». От всех, включая главного режиссера МХАТа Немировича-Данченко, «Батум», несмотря на свои довольно скромные художественные достоинства, имел только благоприятные отзывы (других о произведении, посвященном Сталину, быть не могло). Уже намечались актеры на главные роли. В частности, Сталина должен был играть Н.П. Хмелев.